File size: 468 Bytes
0c6aadc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | [general]
name = "metakernel"
version = 1
edition = 5
license = "Apache-2.0"
backends = [
"cuda",
]
[general.hub]
repo-id = "phanerozoic/metakernel"
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.metakernel]
backend = "cuda"
cuda-capabilities = ["8.0", "8.6", "8.9", "9.0", "10.0", "12.0"]
depends = ["torch"]
src = [
"metakernel_cuda/probes.cu",
"metakernel_cuda/probes_launch.h",
"metakernel_cuda/stamps.cuh",
]
|