Skip to content
KernelIndex
Search⌘K
Sign in

submission 29936

KernelBot user 44 · python · License unknown
Kernel source · 1 lines ↓

Use it

Vendorable · source mirrored · license unknownView source →

No package. Vendor the mirrored source: 1 lines, June 9 Researcher Reciprocity License v1.0.

submission.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-amd-mla-decode-29936?include=source"
interfacepython
Compatibility
measured onAMD Instinct MI300X
declared hardwareAMD Instinct MI300X
architecturesgfx942
dtypesbf16

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
AMD MLA decodebf16 · [128, 1, 7168]
AMD Instinct MI300X
2.02ms
#18 of 79
2025-06-01
Run →

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:fd1d54ad258c889155bab680698d2d54a33be18763772596aec595854ed7644a
license declaredunknown
license concludedunknown
imported2026-08-15

Kernel source

submission.py1 lines
aW1wb3J0IG1hdGgKZnJvbSBkYXRhY2xhc3NlcyBpbXBvcnQgZGF0YWNsYXNzCmltcG9ydCB0b3JjaApmcm9tIHRvcmNoIGltcG9ydCBubgppbXBvcnQgdG9yY2gubm4uZnVuY3Rpb25hbCBhcyBGCmZyb20gdGFzayBpbXBvcnQgaW5wdXRfdCwgb3V0cHV0X3QKCiMgaW1wb3J0IG1sYQoKZGVmIHJvdGF0ZV9oYWxmKHgpOgogICAgeDEgPSB4Wy4uLiwgOiB4LnNoYXBlWy0xXSAvLyAyXQogICAgeDIgPSB4Wy4uLiwgeC5zaGFwZVstMV0gLy8gMiA6XQogICAgcmV0dXJuIHRvcmNoLmNhdCgoLXgyLCB4MSksIGRpbT0tMSkKCkB0b3JjaC5jb21waWxlCmRlZiBhcHBseV9yb3RhcnlfcG9zX2VtYihxLCBrLCBjb3MsIHNpbiwgc2VxX2xlbik6CiAgICBxX2VtYmVkID0gKHEgKiBjb3Nbc2VxX2xlbiAtIDFdKSArIChyb3RhdGVfaGFsZihxKSAqIHNpbltzZXFfbGVuIC0gMV0pCiAgICBrX2VtYmVkID0gKGsgKiBjb3NbOnNlcV9sZW5dKSArIChyb3RhdGVfaGFsZihrKSAqIHNpbls6c2VxX2xlbl0pCiAgICByZXR1cm4gcV9lbWJlZCwga19lbWJlZAoKZGVmIHByZWNvbXB1dGVfcm9wZV9jYWNoZShzZXFfbGVuLCBkaW0sIGRldmljZT0nY3VkYScsIGJhc2U9MTAwMDAuMCk6CiAgICBhc3NlcnQgZGltICUgMiA9PSAwLCAiRW1iZWRkaW5nIGRpbWVuc2lvbiBtdXN0IGJlIGV2ZW4gZm9yIFJvUEUuIgogICAgcG9zaXRpb24gPSB0b3JjaC5hcmFuZ2Uoc2VxX2xlbiwgZGV2aWNlPWRldmljZSwgZHR5cGU9dG9yY2guYmZsb2F0MTYpCiAgICB0aGV0YSA9IGJhc2UgKiogKC10b3JjaC5hcmFuZ2UoMCwgZGltIC8vIDIsIGRldmljZT1kZXZpY2UsIGR0eXBlPXRvcmNoLmJmbG9hdDE2KSAvIChkaW0gLy8gMikpCiAgICBhbmdsZXMgPSB0b3JjaC5vdXRlcihwb3NpdGlvbiwgdGhldGEpCiAgICBvdXQgPSB0b3JjaC5jYXQoW2FuZ2xlcywgYW5nbGVzXSwgZGltPS0xKQogICAgcmV0dXJuIG91dC5jb3MoKSwgb3V0LnNpbigpCgpyb3BlX2NhY2hlID0gcHJlY29tcHV0ZV9yb3BlX2NhY2hlKDYxNDUsIDY0KQoKQHRvcmNoLmNvbXBpbGUKZGVmIGFwcGx5X3NvZnRtYXgocWtfbm9wZSwgcWtfcm9wZSwgbm9wZV9kaW0sIHJvcGVfZGltKToKICAgIHNjb3JlcyA9IChxa19ub3BlICsgcWtfcm9wZSkgLyBtYXRoLnNxcnQocm9wZV9kaW0gKyBub3BlX2RpbSkKICAgIHJldHVybiBGLnNvZnRtYXgoc2NvcmVzLCBkaW09LTEpCgojIGltcG9ydCBvcwojIG9zLnB1dGVudignUFlUT1JDSF9UVU5BQkxFT1BfRU5BQkxFRCcsICcxJykKIyBvcy5wdXRlbnYoJ1BZVE9SQ0hfVFVOQUJMRU9QX1RVTklORycsICcwJykKIyBvcy5wdXRlbnYoJ1BZVE9SQ0hfVFVOQUJMRU9QX1JFQ09SRF9VTlRVTkVEJywgJzEnKQoKIyB0b3JjaC5jdWRhLnR1bmFibGUuZW5hYmxlKCkKCkB0b3JjaC5jb21waWxlCmRlZiBjdXN0b21fa2VybmVsKGRhdGE6IGlucHV0X3QpIC0+IG91dHB1dF90OgogICAgY29uZmlnLCB4LCBrdl9jYWNoZV93cmFwcGVyID0gZGF0YQogICAgcHJlZmlsbCA9IGt2X2NhY2hlX3dyYXBwZXIuc2VxX2xlbgogICAga3ZfY2FjaGUsIHFkb3duLCBxdXAsIGt2ZG93biwga3Z1cCwgd28gPSBrdl9jYWNoZV93cmFwcGVyLmdldF9kYXRhKCksIGNvbmZpZy5RX3Byb2pfZG93bl93ZWlnaHQsIGNvbmZpZy5RX3Byb2pfdXBfd2VpZ2h0LCBjb25maWcuS1ZfcHJval9kb3duX3dlaWdodCwgY29uZmlnLktWX3Byb2pfdXBfd2VpZ2h0LCBjb25maWcud29fd2VpZ2h0CiAgICBvdXRwdXQgPSB0b3JjaC5lbXB0eV9saWtlKHgpCiAgICBiYXRjaF9zaXplLCBzZXFfbGVuLCBuX2hlYWRzID0gY29uZmlnLmJhdGNoX3NpemUsIGNvbmZpZy5zZXFfbGVuLCBjb25maWcubl9oZWFkcwogICAgYXNzZXJ0IHNlcV9sZW4gPT0gMQogICAgbm9wZV9kaW0sIHJvcGVfZGltID0gY29uZmlnLnFrX25vcGVfaGVhZF9kaW0sIGNvbmZpZy5xa19yb3BlX2hlYWRfZGltCiAgICBxX2xvcmFfZGltLCBrdl9sb3JhX2RpbSA9IGNvbmZpZy5xX2xvcmFfcmFuaywgY29uZmlnLmt2X2xvcmFfcmFuawogICAgdl9kaW0gPSBjb25maWcudl9oZWFkX2RpbQogICAgaGlkZGVuX2RpbSA9IGNvbmZpZy5kaW0KICAgICMgc3RlcCAxCiAgICAjIGhpZGRlbiAtPiBrdl9sb3JhWy0xXSAoa3Zkb3duX25vcGUpCiAgICAjIGhpZGRlbiAtPiBrX3JvcGVbLTFdIChrdmRvd25fcm9wZSkKICAgICMga3ZfY2FjaGUgLT4ga3ZfbG9yYSwga19yb3BlCiAgICAjIGhpZGRlbiAtPiBxX2xvcmEgKHFkb3duKQogICAga3ZfY2FjaGU6IHRvcmNoLlRlbnNvciA9IGt2X2NhY2hlWzosIDpwcmVmaWxsICsgc2VxX2xlbl0KICAgIGt2X2NhY2hlWzosIC0xOiwgOl0sIHFfbG9yYSA9IHRvcmNoLmVpbnN1bSgiYiBzIGQsIGwgZCAtPiBiIHMgbCIsIHgsIHRvcmNoLmNvbmNhdCgoa3Zkb3duLCBxZG93biksIGRpbT0wKSkuc3BsaXQoW2t2X2xvcmFfZGltICsgcm9wZV9kaW0sIHFfbG9yYV9kaW1dLCBkaW09LTEpCiAgICBrdl9sb3JhLCBrX3JvcGUgPSBrdl9jYWNoZS5zcGxpdChba3ZfbG9yYV9kaW0sIHJvcGVfZGltXSwgZGltPS0xKQogICAgIyBzdGVwIDIKICAgICMgcV9sb3JhIC0+IHFfcm9wZSAocXVwX3JvcGUpCiAgICAjIHFfbG9yYSAtPiBxX25vcGUgQCBxdXBfbm9wZSBAIGt1cF9ub3BlCiAgICBxdXAgPSBxdXAudmlldyhuX2hlYWRzLCBub3BlX2RpbSArIHJvcGVfZGltLCBxX2xvcmFfZGltKQogICAga3VwX25vcGUsIHZ1cCA9IGt2dXAudmlldyhuX2hlYWRzLCBub3BlX2RpbSArIHZfZGltLCBrdl9sb3JhX2RpbSkuc3BsaXQoW25vcGVfZGltLCB2X2RpbV0sIGRpbT0tMikKICAgIHFfbm9wZSwgcV9yb3BlID0gdG9yY2guZWluc3VtKCJiIHMgbCwgaCBkIGwgLT4gYiBzIGggZCIsIHFfbG9yYSwgcXVwKS5zcGxpdChbbm9wZV9kaW0sIHJvcGVfZGltXSwgZGltPS0xKQogICAgcV9hYnNvcmIgPSB0b3JjaC5laW5zdW0oImIgcyBoIGQsIGggZCBsLT4gYiBzIGggbCIsIHFfbm9wZSwga3VwX25vcGUpCiAgICBxX3JvcGUgPSBxX3JvcGUucGVybXV0ZSgwLCAyLCAxLCAzKQogICAgcV9hYnNvcmIgPSBxX2Fic29yYi5wZXJtdXRlKDAsIDIsIDEsIDMpCiAgICAjIHN0ZXAgMi41IGFwcGx5IHJvcGUKICAgIGNvcywgc2luID0gcm9wZV9jYWNoZQogICAgcV9yb3BlLCBrX3JvcGUgPSBhcHBseV9yb3RhcnlfcG9zX2VtYihxX3JvcGUsIGtfcm9wZSwgY29zLCBzaW4sIHByZWZpbGwgKyBzZXFfbGVuKQogICAgIyBwcmludChmIntxX3JvcGVbMCwgMCwgMCwgLTFdPX0iKQogICAgIyBwcmludChmIntxX25vcGVbMCwgMCwgMCwgLTFdPX0iKQogICAgIyBwcmludChmIntrX3JvcGVbMCwgMSwgMCwgLTFdPX0iKQogICAgIyBwcmludChmIntrX25vcGVbMCwgMCwgMSwgLTFdPX0iKQoKICAgICMgc3RlcCAzCiAgICAjIHFfbm9wZSBAIGt2X2xvcmEgLT4gYXR0bl9ub3BlCiAgICAjIHFfcm9wZSBAIGt2X3JvcGUgLT4gYXR0bl9yb3BlCiAgICAjIHNvZnRtYXgoYXR0bl9ub3BlICsgYXR0bl9yb3BlKSAtPiBhdHRuCiAgICBhdHRuX25vcGUgPSB0b3JjaC5laW5zdW0oImIgaCBzIGwsIGIgcCBsIC0+IGIgaCBzIHAiLCBxX2Fic29yYiwga3ZfbG9yYSkKICAgIGF0dG5fcm9wZSA9IHRvcmNoLmVpbnN1bSgiYiBoIHMgZCwgYiBwIGQgLT4gYiBoIHMgcCIsIHFfcm9wZSwga19yb3BlKQogICAgYXR0ZW50aW9uID0gYXBwbHlfc29mdG1heChhdHRuX25vcGUsIGF0dG5fcm9wZSwgbm9wZV9kaW0sIHJvcGVfZGltKQogICAgIyBzdGVwIDQKICAgICMgYXR0biBAIGt2X2xvcmEgQCB2dXAgQCB3by0+IG91dHB1dAogICAgbyA9IHRvcmNoLmVpbnN1bSgiYiBoIHMgcCwgYiBwIGwgLT4gYiBoIHMgbCIsIGF0dGVudGlvbiwga3ZfbG9yYSkKICAgIG8gPSB0b3JjaC5laW5zdW0oImIgaCBzIGwsIGggdiBsIC0+IGIgaCBzIHYiLCBvLCB2dXApCiAgICB3byA9IHdvLnZpZXcoaGlkZGVuX2RpbSwgbl9oZWFkcywgdl9kaW0pCiAgICBvdXRwdXQgPSB0b3JjaC5laW5zdW0oImIgaCBzIHYsIGQgaCB2IC0+IGIgcyBkIiwgbywgd28pCiAgICAjIG1sYS5tbGFfZGVjb2RlKHgsIGt2X2NhY2hlLmdldF9kYXRhKCksIGNvbmZpZy5RX3Byb2pfZG93bl93ZWlnaHQsIGNvbmZpZy5RX3Byb2pfdXBfd2VpZ2h0LCBjb25maWcuS1ZfcHJval9kb3duX3dlaWdodCwgY29uZmlnLktWX3Byb2pfdXBfd2VpZ2h0LCBjb25maWcud29fd2VpZ2h0LCBvdXRwdXQpCiAgICByZXR1cm4gb3V0cHV0LCBrdl9jYWNoZV93cmFwcGVyLmdldF9kYXRhKCkK

Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0

Best evidence level for this revision: reported

JSON