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