flint-hills-cuda / build.toml
cahlen's picture
CUDA kernel: flint-hills-cuda
42ef9b5 verified
[general]
name = "flint_hills"
universal = false
[torch]
src = ["torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h"]
[kernel.flint_hills]
backend = "cuda"
cuda-capabilities = ["8.0", "9.0", "10.0", "12.0"]
src = ["flint_hills/flint_hills.cu"]
depends = ["torch"]