Skip to content
KernelIndex
Search⌘K

submission 29855

KernelBot user 160 · 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-29855?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
24.8ms
#47 of 79
2025-06-01

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:0cfdb29a8cf6b05e42c66909261616dca67ee6d234312f96be1a21ee3ba7809a
license declaredunknown
license concludedunknown
imported2026-08-15

Kernel source

submission.py1 lines
IyFQT1BDT1JOIGxlYWRlcmJvYXJkIGFtZC1tbGEtZGVjb2RlCiMhUE9QQ09STiBncHVzIE1JMzAwCgppbXBvcnQgbWF0aAppbXBvcnQgdG9yY2gKaW1wb3J0IHRvcmNoLm5uLmZ1bmN0aW9uYWwgYXMgRgpmcm9tIHJlZmVyZW5jZSBpbXBvcnQgZ2VuZXJhdGVfaW5wdXQKCgpkZWYgcHJlY29tcHV0ZV9iYXNpcygKICAgIHNlcV9sZW46IGludCwgaGVhZF9kaW06IGludCwgdGhldGEwOiBmbG9hdCA9IDEwMDAwCikgLT4gdHVwbGVbdG9yY2guVGVuc29yLCB0b3JjaC5UZW5zb3JdOgogICAgZnJlcSA9IHRoZXRhMCAqKiAoCiAgICAgICAgLXRvcmNoLmFyYW5nZSgwLCBoZWFkX2RpbSwgMiwgZHR5cGU9dG9yY2guYmZsb2F0MTYsIGRldmljZT0iY3VkYSIpIC8gaGVhZF9kaW0KICAgICkKICAgIHBvcyA9IHRvcmNoLmFyYW5nZShzZXFfbGVuLCBkdHlwZT10b3JjaC5iZmxvYXQxNiwgZGV2aWNlPSJjdWRhIikKICAgIHRoZXRhID0gdG9yY2gub3V0ZXIocG9zLCBmcmVxKQoKICAgIGNvcyA9IHRvcmNoLmNvcyh0aGV0YSkucmVwZWF0KCgxLCAyKSkKICAgIHNpbiA9IHRvcmNoLnNpbih0aGV0YSkucmVwZWF0KCgxLCAyKSkKCiAgICByZXR1cm4gY29zLCBzaW4KCgpkZWYgcm9wZSgKICAgIHg6IHRvcmNoLlRlbnNvciwgYmFzaXM6IHR1cGxlW3RvcmNoLlRlbnNvciwgdG9yY2guVGVuc29yXSwgcG9zOiBpbnQgPSAwCikgLT4gdG9yY2guVGVuc29yOgogICAgY29zLCBzaW4gPSBiYXNpcwogICAgeDEsIHgyID0geC5jaHVuaygyLCBkaW09LTEpCiAgICByZXR1cm4gKAogICAgICAgIHggKiBjb3NbcG9zIDogcG9zICsgeC5zaGFwZVstMl1dCiAgICAgICAgKyB0b3JjaC5jYXQoKC14MiwgeDEpLCBkaW09LTEpICogc2luW3BvcyA6IHBvcyArIHguc2hhcGVbLTJdXQogICAgKQoKCnJvcGUgPSB0b3JjaC5jb21waWxlKHJvcGUsIG1vZGU9Im1heC1hdXRvdHVuZSIpCgprX2NhY2hlID0gdG9yY2guZW1wdHkoMTI4LCAxMjgsIDYxNDUsIDE5MiwgZHR5cGU9dG9yY2guYmZsb2F0MTYsIGRldmljZT0iY3VkYSIpCnZfY2FjaGUgPSB0b3JjaC5lbXB0eSgxMjgsIDEyOCwgNjE0NSwgMTI4LCBkdHlwZT10b3JjaC5iZmxvYXQxNiwgZGV2aWNlPSJjdWRhIikKCmJhc2lzID0gcHJlY29tcHV0ZV9iYXNpcyg2MTQ1LCA2NCkKCgpkZWYgcHJlY29tcHV0ZV9rdl9wcm9qKCk6CiAgICBnbG9iYWwga19jYWNoZSwgdl9jYWNoZQogICAgCiAgICAjIENvbnN0YW50cwogICAgbmggPSAxMjgKICAgIGRxX25vcGUgPSAxMjgKICAgIGRxX3JvcGUgPSA2NAogICAgZGt2ID0gNTEyCiAgICBka19ub3BlID0gMTI4CiAgICBka19yb3BlID0gNjQKICAgIGR2ID0gMTI4CgogICAgaW5wdXRzID0geydkcSc6IDE1MzYsICdkaW0nOiA3MTY4LCAnYmF0Y2hzaXplJzogMTI4LCAncHJlZmlsbCc6IDYxNDQsICdzZWVkJzogNTI5MX0KICAgIGNvbmZpZywgeCwga3ZfY2FjaGUgPSBnZW5lcmF0ZV9pbnB1dCgqKmlucHV0cykKCiAgICBCLCBDLCBkID0geC5zaGFwZQogICAgcHJldl9sZW4gPSBrdl9jYWNoZS5zZXFfbGVuCiAgICB0b3RhbF9sZW4gPSBwcmV2X2xlbiArIEMKICAgIGludl9zcXJ0X2RrID0gMS4wIC8gbWF0aC5zcXJ0KGRrX25vcGUgKyBkcV9yb3BlKQoKICAgIGt2X2Z1bGwgPSBrdl9jYWNoZS5kYXRhWzosIDpwcmV2X2xlbl0gICMgQiwgdG90YWxfbGVuLCBka3YgKyBka19yb3BlCiAgICBrdl9kb3duX3BhcnQgPSBrdl9mdWxsWy4uLiwgOmRrdl0KICAgIGtfcm9wZV9mdWxsID0ga3ZfZnVsbFsuLi4sIGRrdjpdCiAgICBrdl91cCA9IHRvcmNoLm1tKGt2X2Rvd25fcGFydC5yZXNoYXBlKC0xLCBka3YpLCBjb25maWcuS1ZfcHJval91cF93ZWlnaHQuVCkudmlldygKICAgICAgICBCLCBwcmV2X2xlbiwgbmgsIGRrX25vcGUgKyBkdgogICAgKQogICAga3ZfdXAgPSBrdl91cC50cmFuc3Bvc2UoMSwgMikgICMgQiwgbmgsIHRvdGFsX2xlbiwgZGtfbm9wZSArIGR2CiAgICBrX2NhY2hlWzosIDosIDpwcmV2X2xlbiwgOmRxX25vcGVdID0ga3ZfdXBbLi4uLCA6ZGtfbm9wZV0KICAgIHZfY2FjaGVbOiwgOiwgOnByZXZfbGVuLCA6XSA9IGt2X3VwWy4uLiwgZGtfbm9wZTpdCiAgICBrX3JvcGVfcm90YXRlZCA9IHJvcGUoa19yb3BlX2Z1bGwsIGJhc2lzLCAwKSAgIyBBcHBseSBSb1BFIGZyb20gcG9zaXRpb24gMAogICAga19jYWNoZVs6LCA6LCA6cHJldl9sZW4sIGRxX25vcGU6XSA9IGtfcm9wZV9yb3RhdGVkLnVuc3F1ZWV6ZSgxKS5leHBhbmQoCiAgICAgICAgLTEsIG5oLCAtMSwgLTEKICAgICkKCnByZWNvbXB1dGVfa3ZfcHJvaigpCgpAdG9yY2gubm9fZ3JhZApkZWYgY29tcHV0ZSgKICAgIHg6IHRvcmNoLlRlbnNvciwKICAgIGt2X2NhY2hlOiB0b3JjaC5UZW5zb3IsCiAgICBwcmV2X2xlbjogaW50LAogICAgUV9wcm9qX2Rvd25fd2VpZ2h0OiB0b3JjaC5UZW5zb3IsCiAgICBRX3Byb2pfdXBfd2VpZ2h0OiB0b3JjaC5UZW5zb3IsCiAgICBLVl9wcm9qX2Rvd25fd2VpZ2h0OiB0b3JjaC5UZW5zb3IsCiAgICBLVl9wcm9qX3VwX3dlaWdodDogdG9yY2guVGVuc29yLAogICAgd29fd2VpZ2h0OiB0b3JjaC5UZW5zb3IsCik6CiAgICBnbG9iYWwga19jYWNoZSwgdl9jYWNoZQoKICAgICMgQ29uc3RhbnRzCiAgICBuaCA9IDEyOAogICAgZHFfbm9wZSA9IDEyOAogICAgZHFfcm9wZSA9IDY0CiAgICBka3YgPSA1MTIKICAgIGRrX25vcGUgPSAxMjgKICAgIGRrX3JvcGUgPSA2NAogICAgZHYgPSAxMjgKCiAgICBCLCBDLCBkID0geC5zaGFwZQogICAgdG90YWxfbGVuID0gcHJldl9sZW4gKyBDCiAgICBpbnZfc3FydF9kayA9IDEuMCAvIG1hdGguc3FydChka19ub3BlICsgZHFfcm9wZSkKCiAgICBrX2NhY2hlLnJlc2l6ZV8oMTI4LCAxMjgsIHRvdGFsX2xlbiwgMTkyKQogICAgdl9jYWNoZS5yZXNpemVfKDEyOCwgMTI4LCB0b3RhbF9sZW4sIDEyOCkKCiAgICBxX2Rvd24gPSB0b3JjaC5tbSh4LnZpZXcoLTEsIGQpLCBRX3Byb2pfZG93bl93ZWlnaHQuVCkgICMgKEIqQywgUV9kb3duX2RpbSkKICAgIHEgPSB0b3JjaC5tbShxX2Rvd24sIFFfcHJval91cF93ZWlnaHQuVCkudmlldyhCLCBDLCBuaCwgZHFfbm9wZSArIGRxX3JvcGUpCiAgICBxID0gcS50cmFuc3Bvc2UoMSwgMikgICMgQiwgbmgsIEMsIGRxX25vcGUgKyBkcV9yb3BlCiAgICBxX3JvcGUgPSBxWy4uLiwgZHFfbm9wZTpdCiAgICBxWzosIDosIDosIGRxX25vcGU6XSA9IHJvcGUocV9yb3BlLCBiYXNpcywgcHJldl9sZW4pCgogICAga3ZfZG93biA9IHRvcmNoLm1tKHgudmlldygtMSwgZCksIEtWX3Byb2pfZG93bl93ZWlnaHQuVCkudmlldyhCLCBDLCAtMSkKICAgIGt2X2NhY2hlWzosIHByZXZfbGVuOnRvdGFsX2xlbl0gPSBrdl9kb3duCgogICAga3ZfZnVsbCA9IGt2X2NhY2hlWzosIDp0b3RhbF9sZW5dICAjIEIsIHRvdGFsX2xlbiwgZGt2ICsgZGtfcm9wZQogICAga3ZfZG93bl9wYXJ0ID0ga3ZfZnVsbFsuLi4sIDpka3ZdCiAgICBrX3JvcGVfZnVsbCA9IGt2X2Z1bGxbLi4uLCBka3Y6XQogICAga3ZfdXAgPSB0b3JjaC5tbShrdl9kb3duX3BhcnQucmVzaGFwZSgtMSwgZGt2KSwgS1ZfcHJval91cF93ZWlnaHQuVCkudmlldygKICAgICAgICBCLCB0b3RhbF9sZW4sIG5oLCBka19ub3BlICsgZHYKICAgICkKICAgIGt2X3VwID0ga3ZfdXAudHJhbnNwb3NlKDEsIDIpICAjIEIsIG5oLCB0b3RhbF9sZW4sIGRrX25vcGUgKyBkdgogICAga19ub3BlID0ga3ZfdXBbLi4uLCA6ZGtfbm9wZV0KICAgIHYgPSBrdl91cFsuLi4sIGRrX25vcGU6XQogICAga19yb3BlX3JvdGF0ZWQgPSByb3BlKGtfcm9wZV9mdWxsLCBiYXNpcywgMCkgICMgQXBwbHkgUm9QRSBmcm9tIHBvc2l0aW9uIDAKICAgIGtfcm9wZV9leHBhbmRlZCA9IGtfcm9wZV9yb3RhdGVkLnVuc3F1ZWV6ZSgxKS5leHBhbmQoLTEsIG5oLCAtMSwgLTEpCiAgICBrID0gdG9yY2guY2F0KFtrX25vcGUsIGtfcm9wZV9leHBhbmRlZF0sIGRpbT0tMSkKCiAgICBCX25oID0gQiAqIG5oCiAgICBxX2ZsYXQgPSBxLnZpZXcoQl9uaCwgQywgLTEpCiAgICBrX2ZsYXQgPSBrLnZpZXcoQl9uaCwgdG90YWxfbGVuLCAtMSkKICAgIHZfZmxhdCA9IHYucmVzaGFwZShCX25oLCB0b3RhbF9sZW4sIGR2KQoKICAgIHNjb3JlcyA9IHRvcmNoLmJtbShxX2ZsYXQsIGtfZmxhdC50cmFuc3Bvc2UoLTEsIC0yKSkKICAgIHNjb3Jlcy5tdWxfKGludl9zcXJ0X2RrKQogICAgYXR0bl93ZWlnaHRzID0gRi5zb2Z0bWF4KHNjb3JlcywgZGltPS0xKQogICAgeV9mbGF0ID0gdG9yY2guYm1tKGF0dG5fd2VpZ2h0cywgdl9mbGF0KSAgIyBCX25oLCBDLCBkdgogICAgeSA9IHlfZmxhdC5yZXNoYXBlKEIsIG5oLCBDLCBkdikudHJhbnNwb3NlKDEsIDIpLnJlc2hhcGUoQiwgQywgLTEpCiAgICBvdXQgPSB0b3JjaC5tbSh5LnZpZXcoLTEsIG5oICogZHYpLCB3b193ZWlnaHQuVCkudmlldyhCLCBDLCAtMSkKCiAgICByZXR1cm4gb3V0CgpAdG9yY2gubm9fZ3JhZApkZWYgY29tcHV0ZV9yYW5rZWQoCiAgICB4OiB0b3JjaC5UZW5zb3IsCiAgICBrdl9jYWNoZTogdG9yY2guVGVuc29yLAogICAgcHJldl9sZW46IGludCwKICAgIFFfcHJval9kb3duX3dlaWdodDogdG9yY2guVGVuc29yLAogICAgUV9wcm9qX3VwX3dlaWdodDogdG9yY2guVGVuc29yLAogICAgS1ZfcHJval9kb3duX3dlaWdodDogdG9yY2guVGVuc29yLAogICAgS1ZfcHJval91cF93ZWlnaHQ6IHRvcmNoLlRlbnNvciwKICAgIHdvX3dlaWdodDogdG9yY2guVGVuc29yLAopOgogICAgZ2xvYmFsIGtfY2FjaGUsIHZfY2FjaGUKCiAgICAjIENvbnN0YW50cwogICAgbmggPSAxMjgKICAgIGRxX25vcGUgPSAxMjgKICAgIGRxX3JvcGUgPSA2NAogICAgZGt2ID0gNTEyCiAgICBka19ub3BlID0gMTI4CiAgICBka19yb3BlID0gNjQKICAgIGR2ID0gMTI4CgogICAgQiwgQywgZCA9IHguc2hhcGUKICAgIHRvdGFsX2xlbiA9IHByZXZfbGVuICsgQwogICAgaW52X3NxcnRfZGsgPSAxLjAgLyBtYXRoLnNxcnQoZGtfbm9wZSArIGRxX3JvcGUpCgogICAga19jYWNoZS5yZXNpemVfKDEyOCwgMTI4LCB0b3RhbF9sZW4sIDE5MikKICAgIHZfY2FjaGUucmVzaXplXygxMjgsIDEyOCwgdG90YWxfbGVuLCAxMjgpCgogICAgcV9kb3duID0gdG9yY2gubW0oeC52aWV3KC0xLCBkKSwgUV9wcm9qX2Rvd25fd2VpZ2h0LlQpICAjIChCKkMsIFFfZG93bl9kaW0pCiAgICBxID0gdG9yY2gubW0ocV9kb3duLCBRX3Byb2pfdXBfd2VpZ2h0LlQpLnZpZXcoQiwgQywgbmgsIGRxX25vcGUgKyBkcV9yb3BlKQogICAgcSA9IHEudHJhbnNwb3NlKDEsIDIpICAjIEIsIG5oLCBDLCBkcV9ub3BlICsgZHFfcm9wZQogICAgcV9yb3BlID0gcVsuLi4sIGRxX25vcGU6XQogICAgcVs6LCA6LCA6LCBkcV9ub3BlOl0gPSByb3BlKHFfcm9wZSwgYmFzaXMsIHByZXZfbGVuKQoKICAgIGt2X2Rvd24gPSB0b3JjaC5tbSh4LnZpZXcoLTEsIGQpLCBLVl9wcm9qX2Rvd25fd2VpZ2h0LlQpLnZpZXcoQiwgQywgLTEpCiAgICBrdl9jYWNoZVs6LCBwcmV2X2xlbjp0b3RhbF9sZW5dID0ga3ZfZG93bgoKICAgIGt2X2Z1bGwgPSBrdl9kb3duCiAgICBrdl9kb3duX3BhcnQgPSBrdl9mdWxsWy4uLiwgOmRrdl0KICAgIGtfcm9wZV9mdWxsID0ga3ZfZnVsbFsuLi4sIGRrdjpdCiAgICBrdl91cCA9IHRvcmNoLm1tKGt2X2Rvd25fcGFydC5yZXNoYXBlKC0xLCBka3YpLCBLVl9wcm9qX3VwX3dlaWdodC5UKS52aWV3KAogICAgICAgIEIsIDEsIG5oLCBka19ub3BlICsgZHYKICAgICkKICAgIGt2X3VwID0ga3ZfdXAudHJhbnNwb3NlKDEsIDIpICAjIEIsIG5oLCB0b3RhbF9sZW4sIGRrX25vcGUgKyBkdgogICAga19ub3BlID0ga3ZfdXBbLi4uLCA6ZGtfbm9wZV0KICAgIHZfbGFzdCA9IGt2X3VwWy4uLiwgZGtfbm9wZTpdCiAgICBrX3JvcGVfcm90YXRlZCA9IHJvcGUoa19yb3BlX2Z1bGwsIGJhc2lzLCBwcmV2X2xlbikgICMgQXBwbHkgUm9QRSBmcm9tIHBvc2l0aW9uIDAKICAgIGtfcm9wZV9leHBhbmRlZCA9IGtfcm9wZV9yb3RhdGVkLnVuc3F1ZWV6ZSgxKS5leHBhbmQoLTEsIG5oLCAtMSwgLTEpCiAgICBrX2xhc3QgPSB0b3JjaC5jYXQoW2tfbm9wZSwga19yb3BlX2V4cGFuZGVkXSwgZGltPS0xKQoKICAgIGssIHYgPSBrX2NhY2hlLCB2X2NhY2hlCiAgICBrWzosIDosIHByZXZfbGVuOnRvdGFsX2xlbl0gPSBrX2xhc3QKICAgIHZbOiwgOiwgcHJldl9sZW46dG90YWxfbGVuXSA9IHZfbGFzdAoKICAgIEJfbmggPSBCICogbmgKICAgIHFfZmxhdCA9IHEudmlldyhCX25oLCBDLCAtMSkKICAgIGtfZmxhdCA9IGsudmlldyhCX25oLCB0b3RhbF9sZW4sIC0xKQogICAgdl9mbGF0ID0gdi52aWV3KEJfbmgsIHRvdGFsX2xlbiwgZHYpCgogICAgc2NvcmVzID0gdG9yY2guYm1tKHFfZmxhdCwga19mbGF0LnRyYW5zcG9zZSgtMSwgLTIpKQogICAgc2NvcmVzLm11bF8oaW52X3NxcnRfZGspCiAgICBhdHRuX3dlaWdodHMgPSBGLnNvZnRtYXgoc2NvcmVzLCBkaW09LTEpCiAgICB5X2ZsYXQgPSB0b3JjaC5ibW0oYXR0bl93ZWlnaHRzLCB2X2ZsYXQpICAjIEJfbmgsIEMsIGR2CiAgICB5ID0geV9mbGF0LnJlc2hhcGUoQiwgbmgsIEMsIGR2KS50cmFuc3Bvc2UoMSwgMikucmVzaGFwZShCLCBDLCAtMSkKICAgIG91dCA9IHRvcmNoLm1tKHkudmlldygtMSwgbmggKiBkdiksIHdvX3dlaWdodC5UKS52aWV3KEIsIEMsIC0xKQoKICAgIHJldHVybiBvdXQKCgojIGNvbXB1dGUgPSB0b3JjaC5jb21waWxlKGNvbXB1dGUsIG1vZGU9Im1heC1hdXRvdHVuZSIpCgoKZGVmIGN1c3RvbV9rZXJuZWwoZGF0YTogdHVwbGUpIC0+IHR1cGxlOgogICAgY29uZmlnLCB4LCBrdl9jYWNoZSA9IGRhdGEKICAgIHByZXZfbGVuID0ga3ZfY2FjaGUuc2VxX2xlbgogICAgY29tcHV0ZV9mbiA9IGNvbXB1dGVfcmFua2VkIGlmIGt2X2NhY2hlLnNlcV9sZW4gPT0gNjE0NCBlbHNlIGNvbXB1dGUKICAgIG91dCA9IGNvbXB1dGVfZm4oCiAgICAgICAgeCwKICAgICAgICBrdl9jYWNoZS5kYXRhLAogICAgICAgIHByZXZfbGVuLAogICAgICAgIGNvbmZpZy5RX3Byb2pfZG93bl93ZWlnaHQsCiAgICAgICAgY29uZmlnLlFfcHJval91cF93ZWlnaHQsCiAgICAgICAgY29uZmlnLktWX3Byb2pfZG93bl93ZWlnaHQsCiAgICAgICAgY29uZmlnLktWX3Byb2pfdXBfd2VpZ2h0LAogICAgICAgIGNvbmZpZy53b193ZWlnaHQsCiAgICApCiAgICByZXR1cm4gb3V0LCBrdl9jYWNoZS5kYXRhCg==

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

Best evidence level for this revision: reported

JSON