File size: 289 Bytes
55ce07b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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",
]