Skip to content
KernelIndex
Search⌘K

submission 544990

rajesh0042 · python · License unknown

Use it

Vendorable · source mirrored · license unknownView source →

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

sort_simple.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-sort-v2-544990?include=source"
interfacepython
Compatibility
measured onNVIDIA H100
declared hardwareNVIDIA H100
architecturessm_90
dtypesfp32

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
Sortsuite of 5 cases
NVIDIA H100
6.54ms
#12 of 26
2026-03-13

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:e958f84ef901f3d0a9482766d16c846c7f064b8f0eff3c2c8d3df41c21a451e4
license declaredunknown
license concludedunknown
authorsrajesh0042
imported2026-08-15

Kernel source

sort_simple.py11 lines
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:
    data, output = data
    torch.sort(data, out=(output, torch.empty_like(data, dtype=torch.long)))
    return output

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

Best evidence level for this revision: reported

JSON