submission 34494
mav · python · License unknown
Use it
Vendorable · source mirrored · license unknownView source →
No package. Vendor the mirrored source: 20 lines, June 9 Researcher Reciprocity License v1.0.
vectoradd.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-vectoradd-v2-34494?include=source"interfacepython
Compatibility
measured onNVIDIA A100
declared hardwareNVIDIA A100
architecturessm_80
dtypesfp16
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:b7f748fc1474f4d1c836e2ede5b2cca0135818712fdc28945ea5986dd6c5692b
license declaredunknown
license concludedunknown
authorsmav
imported2026-08-15
Kernel source
vectoradd.py20 lines
from task import input_t, output_t
import jax
import jax.numpy as jnp
import torch
@jax.jit
def add(A, B):
return A + B
def custom_kernel(data: input_t) -> output_t:
(A, B, C) = data
A = jnp.asarray(A)
B = jnp.asarray(B)
C = add(A, B)
C_torch = torch.utils.dlpack.from_dlpack(jax.dlpack.to_dlpack(C))
return C_torch
Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0
Best evidence level for this revision: reported
JSON