File size: 421 Bytes
12075d1 6a2f5d2 12075d1 6a2f5d2 12075d1 6a2f5d2 12075d1 | 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 | [general]
name = "cv_utils"
universal = false
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.cv_utils]
depends = ["torch"]
backend = "cuda"
src = [
"cv_utils/connected_components.cu",
"cv_utils/generic_nms.cu",
]
cuda-flags = [
"-DCUDA_HAS_FP16=1",
"-D__CUDA_NO_HALF_OPERATORS__",
"-D__CUDA_NO_HALF_CONVERSIONS__",
"-D__CUDA_NO_HALF2_OPERATORS__",
]
|