submission 539666
Jaber Jaber · python · License unknown
Use it
Vendorable · source mirrored · license unknownView source →
No package. Vendor the mirrored source: 20 lines, June 9 Researcher Reciprocity License v1.0.
submission_cublas.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-matmul-v2-539666?include=source"interfacepython
Compatibility
measured onNVIDIA B200
declared hardwareNVIDIA B200
architecturessm_100
dtypesfp16
Benchmark evidence
1 measurement across 1 GPU, fastest first.
Operation / workload
Hardware
Latency
Rank
Observed
Reported · How evidence levels are derived →
Source and license
sourceavailable
revision digestsha256:5e30cb791a62681a27cc08367c05c48240db9a03a0447760bac91887ef28f75b
license declaredunknown
license concludedunknown
authorsJaber Jaber
imported2026-08-15
Kernel source
submission_cublas.py20 lines
"""
Matmul v2 -- cuBLAS via torch.mm with pre-allocated output.
Advantages over reference:
- No DeterministicContext overhead (reference uses it)
- No output allocation (write to pre-allocated c)
- Directly calls cuBLAS cublasGemmEx with tensor cores
"""
import os
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
import torch
from task import input_t, output_t
def custom_kernel(data: input_t) -> output_t:
a, b, c = data
torch.mm(a, b, out=c)
return c
Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0
Best evidence level for this revision: reported
JSON