diff --git a/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu b/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu index bd80adad77e24f86bf04fc6e6db9181928266d39..5edd28b325496fca1eb80f579c18a99bdbcfda71 100644 --- a/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu +++ b/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu @@ -6,6 +6,8 @@ #include <vector> +#define RESTRICT __restrict__ + {% for g in cuda_globals -%} {{ g }} {% endfor %}