File size: 283 Bytes
7ec6a18 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | [general]
name = "class_numbers"
universal = false
[torch]
src = ["torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h"]
[kernel.class_numbers]
backend = "cuda"
cuda-capabilities = ["8.0", "9.0", "10.0", "12.0"]
src = ["class_numbers/class_numbers_v2.cu"]
depends = ["torch"]
|