drbh
/

img2gray

Model card Files Files and versions

img2gray / build.toml

drbh

feat: small grayscale kernel

846d481 5 months ago

217 Bytes

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