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