Kernels:
Trusted publisher
File size: 289 Bytes
64e39bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | [general]
name = "mra"
universal = false
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/cuda_launch.h",
]
[kernel.mra]
backend = "cuda"
depends = ["torch"]
src = [
"mra/cuda_kernel.cu",
"mra/cuda_kernel.h",
"mra/cuda_launch.cu",
"mra/cuda_launch.h",
] |