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 %}