build.toml

[general]
name = "img2gray"
universal = false

[torch]
src = [
  "torch-ext/torch_binding.cpp",
  "torch-ext/torch_binding.h"
]

[kernel.img2gray]
backend = "cuda"
depends = ["torch"]
src = [
    "csrc/img2gray.cu",
]