[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"]