Skip to content
KernelIndex
Search⌘K
Sign in

submission 554690

idanbeck · python · License unknown
Kernel source · 21 lines ↓

Use it

Vendorable · source mirrored · license unknownView source →

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

popcorn_conv2d_v2_probe_benchmark_true.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-conv2d-v2-554690?include=source"
interfacepython
Compatibility
measured onNVIDIA A100
declared hardwareNVIDIA A100
architecturessm_80
dtypesfp32

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
2D convolutionsuite of 5 cases
NVIDIA A100
20.2ms
#14 of 40
2026-03-14
Run →

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:9821507b947536893bebcda35f26640f2f8989d2ccb2db8ea874e5bc4dd45089
license declaredunknown
license concludedunknown
authorsidanbeck
imported2026-08-15

Kernel source

popcorn_conv2d_v2_probe_benchmark_true.py21 lines
#!POPCORN leaderboard conv2d_v2
#!POPCORN gpu A100
from task import input_t, output_t
import torch
import torch.nn.functional as F

@torch.inference_mode()
def custom_kernel(data: input_t) -> output_t:
    x, w, out = data
    torch.use_deterministic_algorithms(True)
    torch.set_float32_matmul_precision('highest')
    torch.backends.cudnn.benchmark = True
    torch.backends.cudnn.deterministic = True
    torch.backends.cuda.matmul.allow_tf32 = False
    torch.backends.cudnn.allow_tf32 = False
    x_cl = x.contiguous(memory_format=torch.channels_last)
    w_cl = w.contiguous(memory_format=torch.channels_last)
    y = F.conv2d(x_cl, w_cl, stride=1, padding=0)
    out[...] = y.contiguous()
    return out

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 554642.

  ⋯ 8 unchanged lines
      x, w, out = data
      torch.use_deterministic_algorithms(True)
      torch.set_float32_matmul_precision('highest')
-     torch.backends.cudnn.benchmark = False
+     torch.backends.cudnn.benchmark = True
      torch.backends.cudnn.deterministic = True
      torch.backends.cuda.matmul.allow_tf32 = False
      torch.backends.cudnn.allow_tf32 = False

Best evidence level for this revision: reported

JSON