Skip to content
KernelIndex
Search⌘K

submission 704454

stashuk-olek · python · License unknown

Use it

Vendorable · source mirrored · license unknownView source →

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

submission.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-conv2d-v2-704454?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
2D convolutionsuite of 5 cases
NVIDIA H100
75.3ms
#10 of 35
2026-04-02

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:cff8b1a5e7fe7bcdae422dadf55cb018b202e8768c7352a0546cb01eb00524b0
license declaredunknown
license concludedunknown
authorsstashuk-olek
imported2026-08-15

Kernel source

submission.py17 lines
#!POPCORN leaderboard conv2d_v2
#!POPCORN gpu H100

import torch
import torch.nn.functional as F
from task import input_t, output_t

# Disable TF32: H100 enables it by default, but eval.py reference uses exact float32
torch.backends.cudnn.allow_tf32 = False
torch.backends.cuda.matmul.allow_tf32 = False


def custom_kernel(data: input_t) -> output_t:
    input_tensor, kernel, output = data
    output[...] = F.conv2d(input_tensor, kernel, stride=1, padding=0)
    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