cahlen's picture
CUDA kernel: lyapunov-spectrum-cuda
3e3c11a verified
Raw
History Blame Contribute Delete
296 Bytes
[general]
name = "lyapunov_spectrum"
universal = false
[torch]
src = ["torch-ext/torch_binding.cpp", "torch-ext/torch_binding.h"]
[kernel.lyapunov_spectrum]
backend = "cuda"
cuda-capabilities = ["8.0", "9.0", "10.0", "12.0"]
src = ["lyapunov_spectrum/lyapunov_spectrum.cu"]
depends = ["torch"]