submission 545053
rajesh0042 · python · License unknown
Use it
Vendorable · source mirrored · license unknownView source →
No package. Vendor the mirrored source: 19 lines, June 9 Researcher Reciprocity License v1.0.
conv2d_v2.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-conv2d-v2-545053?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
Reported · How evidence levels are derived →
Source and license
sourceavailable
revision digestsha256:c2737a586a9a2ce5daf2644407f50b3621c8ccacb8819c8dfad0d0d40cf52b11
license declaredunknown
license concludedunknown
authorsrajesh0042
imported2026-08-15
Kernel source
conv2d_v2.py19 lines
import os
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
import torch
import torch.nn.functional as F
from task import input_t, output_t
# Force deterministic behavior to match reference exactly
torch.backends.cudnn.allow_tf32 = False
torch.backends.cuda.matmul.allow_tf32 = False
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
def custom_kernel(data: input_t) -> output_t:
input_tensor, kernel, output = data
result = F.conv2d(input_tensor, kernel, stride=1, padding=0)
output[...] = result
return output
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 545037.
Best evidence level for this revision: reported
JSON