submission 71005
Jaber · python · License unknown
Use it
Vendorable · source mirrored · license unknownView source →
No package. Vendor the mirrored source: 7 lines, June 9 Researcher Reciprocity License v1.0.
submission.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-sort-v2-71005?include=source"interfacepython
Compatibility
measured onNVIDIA H100
declared hardwareNVIDIA H100
architecturessm_90
dtypesfp32
Benchmark evidence
1 measurement across 1 GPU, fastest first.
Reported · How evidence levels are derived →
Source and license
sourceavailable
revision digestsha256:6c0bf39954b42945b52290eee796018a1d16153b70886451c353360830a36f8c
license declaredunknown
license concludedunknown
authorsJaber
imported2026-08-15
Kernel source
submission.py7 lines
import torch
def custom_kernel(data):
input_tensor, output_tensor = data
output_tensor.copy_(torch.sort(input_tensor, stable=False)[0])
return output_tensor
Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0
Changes from previous submission
Against this author's previous submission submission 70989.
import torch- try:- import cupy as cp- HAS_CUPY = True- except ImportError:- HAS_CUPY = False--def custom_kernel(data):input_tensor, output_tensor = data-- if HAS_CUPY:- cupy_input = cp.from_dlpack(input_tensor)- cupy_sorted = cp.sort(cupy_input)- output_tensor.copy_(torch.from_dlpack(cupy_sorted))- else:- output_tensor[...] = torch.sort(input_tensor)[0]-+ output_tensor.copy_(torch.sort(input_tensor, stable=False)[0])return output_tensor
Best evidence level for this revision: reported
JSON