Skip to content
KernelIndex
Search⌘K

submission 26172

KernelBot user 126 · python · License unknown

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-26172?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
112.1ms
#49 of 79
2025-05-22

Reported · How evidence levels are derived →

Source and license

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

Kernel source

submission.py1 lines
aW1wb3J0IG1hdGgKCmltcG9ydCB0b3JjaAppbXBvcnQgdG9yY2gubm4uZnVuY3Rpb25hbCBhcyBGCgpmcm9tIHRhc2sgaW1wb3J0IGlucHV0X3QsIG91dHB1dF90CgoKZGVmIHJvdGF0ZV9oYWxmKHg6IHRvcmNoLlRlbnNvcikgLT4gdG9yY2guVGVuc29yOgogICAgeDEsIHgyID0geC5jaHVuaygyLCBkaW09LTEpCiAgICByZXR1cm4gdG9yY2guY2F0KCgteDIsIHgxKSwgZGltPS0xKQoKZGVmIHJvcGUoeDogdG9yY2guVGVuc29yLCB0aGV0YTogdG9yY2guVGVuc29yICxzdGFydF9wb3M6IGludCA9IDApIC0+IHRvcmNoLlRlbnNvcjoKICAgIHNlcV9sZW4gPSB4LnNpemUoLTIpCiAgICBkX21vZGVsID0geC5zaXplKC0xKQogICAgc2VxX2lkeCA9IHRvcmNoLmFyYW5nZShzdGFydF9wb3MsIHN0YXJ0X3BvcyArIHNlcV9sZW4sIGRldmljZT14LmRldmljZSkKICAgIGlkeF90aGV0YSA9IHRvcmNoLmVpbnN1bSgncyxkLT5zZCcsIHNlcV9pZHgsIHRoZXRhKQogICAgaWR4X3RoZXRhMiA9IHRvcmNoLmNhdChbaWR4X3RoZXRhLCBpZHhfdGhldGFdLCBkaW09LTEpCiAgICBjb3MgPSBpZHhfdGhldGEyLmNvcygpLnRvKHRvcmNoLmJmbG9hdDE2KQogICAgc2luID0gaWR4X3RoZXRhMi5zaW4oKS50byh0b3JjaC5iZmxvYXQxNikKICAgIHJldHVybiB4ICogY29zICsgcm90YXRlX2hhbGYoeCkgKiBzaW4KCgpkZWYgY3VzdG9tX2tlcm5lbChkYXRhOiBpbnB1dF90KSAtPiBvdXRwdXRfdDoKICAgIGNvbmZpZywgeCwga3ZfY2FjaGUgPSBkYXRhCgogICAgdGhldGEgPSAxMDAwMCAqKiAoLXRvcmNoLmFyYW5nZSgwLCBjb25maWcucWtfcm9wZV9oZWFkX2RpbS8vMixkdHlwZT10b3JjaC5iZmxvYXQxNikgLyAoY29uZmlnLnFrX3JvcGVfaGVhZF9kaW0vLzIpKS5jdWRhKCkKCiAgICAjIHNlcV9sZW4gPSAxIGFsd2F5cyBoZXJlCiAgICBiYXRjaF9zaXplLCBzZXFfbGVuLCBtb2RlbF9kaW0gPSB4LnNpemUoKQoKICAgICMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjCiAgICAjICAgICAgICAgICAgICAgICBTdGVwIDE6IEhhbmRsZSBkb3duLXByb2plY3Rpb24gKyBLViBjYWNoZSAgICAgICAgICAgICAgICAgICAgIwogICAgIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMKCiAgICBxX2xvcmEgPSBGLmxpbmVhcih4LCBjb25maWcuUV9wcm9qX2Rvd25fd2VpZ2h0KQogICAga3ZfbG9yYSA9IEYubGluZWFyKHgsIGNvbmZpZy5LVl9wcm9qX2Rvd25fd2VpZ2h0KQogICAga3ZfbG9yYSwga3ZfbGVuID0ga3ZfY2FjaGUoa3ZfbG9yYSkKICAgIHF1ZXJ5X3BvcyA9IGt2X2xlbiAtIDEKCiAgICAjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIwogICAgIyAgICAgICAgICAgICAgICAgIFN0ZXAgMjogVXAtcHJvamVjdCBhbmQgcHJlcGFyZSBOb1BFICsgUm9QRSAgICAgICAgICAgICAgICAgICMKICAgICMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjCgogICAgIyBIYW5kbGUgcXVlcmllcyBRIGZpcnN0CiAgICBxX25vcGVfYW5kX3JvcGUgPSBGLmxpbmVhcihxX2xvcmEsIGNvbmZpZy5RX3Byb2pfdXBfd2VpZ2h0KS52aWV3KAogICAgICAgIGJhdGNoX3NpemUsIHNlcV9sZW4sIGNvbmZpZy5uX2hlYWRzLCBjb25maWcucWtfbm9wZV9oZWFkX2RpbSArIGNvbmZpZy5xa19yb3BlX2hlYWRfZGltKQogICAgcV9ub3BlLCBxX3JvcGUgPSB0b3JjaC5zcGxpdChxX25vcGVfYW5kX3JvcGUsIFtjb25maWcucWtfbm9wZV9oZWFkX2RpbSwgY29uZmlnLnFrX3JvcGVfaGVhZF9kaW1dLCBkaW09LTEpCgogICAgIyBIYW5kbGUga2V5cyBhbmQgdmFsdWVzIEsvVi4gViBkb2VzIG5vdCBuZWVkIFJvUEUKICAgIGt2X25vcGUsIGtfcm9wZSA9IHRvcmNoLnNwbGl0KGt2X2xvcmEsIFtjb25maWcua3ZfbG9yYV9yYW5rLCBjb25maWcucWtfcm9wZV9oZWFkX2RpbV0sIGRpbT0tMSkKICAgIGt2X25vcGUgPSBGLmxpbmVhcihrdl9ub3BlLCBjb25maWcuS1ZfcHJval91cF93ZWlnaHQpLnZpZXcoCiAgICAgICAgYmF0Y2hfc2l6ZSwga3ZfbGVuLCBjb25maWcubl9oZWFkcywgY29uZmlnLnFrX25vcGVfaGVhZF9kaW0gKyBjb25maWcudl9oZWFkX2RpbSkKICAgIGtfbm9wZSwgdiA9IHRvcmNoLnNwbGl0KGt2X25vcGUsIFtjb25maWcucWtfbm9wZV9oZWFkX2RpbSwgY29uZmlnLnZfaGVhZF9kaW1dLCBkaW09LTEpCgogICAgIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMKICAgICMgICAgICAgICAgICAgICAgICAgIFN0ZXAgMzogSGFuZGxlIFJvUEUgU3RyZWFtICAgICAgICAgICAgICAgICAgICAgICAgICAgICAgICAjCiAgICAjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIwoKICAgICMgQ29tcHV0ZSBSb1BFIGZvciBxdWVyaWVzIGFuZCBjb21iaW5lIHdpdGggbm8tUm9QRSBwYXJ0CiAgICBxX3JvcGUgPSBxX3JvcGUucmVzaGFwZShiYXRjaF9zaXplLCBjb25maWcubl9oZWFkcywgc2VxX2xlbiwgY29uZmlnLnFrX3JvcGVfaGVhZF9kaW0pCiAgICBxX3JvcGUgPSByb3BlKHFfcm9wZSwgdGhldGEsIHN0YXJ0X3Bvcz1xdWVyeV9wb3MpCiAgICBxX3JvcGUgPSBxX3JvcGUucmVzaGFwZShiYXRjaF9zaXplLCBzZXFfbGVuLCBjb25maWcubl9oZWFkcywgY29uZmlnLnFrX3JvcGVfaGVhZF9kaW0pCiAgICBxID0gdG9yY2guY29uY2F0KFtxX25vcGUsIHFfcm9wZV0sIGRpbT0tMSkKCiAgICAjIENvbXB1dGUgUm9QRSBmb3Iga2V5cyBhbmQgY29tYmluZSB3aXRoIG5vLVJvUEUgcGFydAogICAga19yb3BlID0ga19yb3BlWzosIDosIE5vbmUsIDpdCiAgICBrX3JvcGUgPSByb3BlKGtfcm9wZSwgdGhldGEpLmV4cGFuZCgtMSwtMSxjb25maWcubl9oZWFkcywtMSkKICAgIGsgPSB0b3JjaC5jb25jYXQoW2tfbm9wZSwga19yb3BlXSwgZGltPS0xKQoKICAgICMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjCiAgICAjICAgICAgICAgICAgICAgICAgICAgICAgQ29tcHV0ZSBNdWx0aS1oZWFkIEF0dGVudGlvbiAgICAgICAgICAgICAgICAgICAgICAgICAgIwogICAgIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMKCiAgICBxID0gcS5yZXNoYXBlKGJhdGNoX3NpemUsIGNvbmZpZy5uX2hlYWRzLCAxLCBjb25maWcucWtfcm9wZV9oZWFkX2RpbSArIGNvbmZpZy5xa19ub3BlX2hlYWRfZGltKQogICAgayA9IGsucmVzaGFwZShiYXRjaF9zaXplLCBjb25maWcubl9oZWFkcywga3ZfbGVuLCBjb25maWcucWtfcm9wZV9oZWFkX2RpbSArIGNvbmZpZy5xa19ub3BlX2hlYWRfZGltKQogICAgdiA9IHYucmVzaGFwZShiYXRjaF9zaXplLCBjb25maWcubl9oZWFkcywgLTEsIGNvbmZpZy52X2hlYWRfZGltKQogICAgc2NvcmVzID0gdG9yY2gubWF0bXVsKHEsIGsudHJhbnNwb3NlKC0xLCAtMikpIC8gbWF0aC5zcXJ0KGNvbmZpZy5xa19yb3BlX2hlYWRfZGltICsgY29uZmlnLnFrX25vcGVfaGVhZF9kaW0pCiAgICBhdHRuID0gRi5zb2Z0bWF4KHNjb3JlcywgZGltPS0xKS50byh0b3JjaC5iZmxvYXQxNikKICAgIHkgPSB0b3JjaC5tYXRtdWwoYXR0biwgdikudmlldyhiYXRjaF9zaXplLCAxLCAtMSkKICAgIHkgPSBGLmxpbmVhcih5LCBjb25maWcud29fd2VpZ2h0KQoKICAgIHJldHVybiB5LCBrdl9jYWNoZS5nZXRfZGF0YSgpCg==

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

Best evidence level for this revision: reported

JSON