From 92db4264a4f79c17d3545a20b8566f38dddbf70d Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 7 Aug 2019 14:26:52 +0200
Subject: [PATCH] Add defintion of RESTRICT to torch CUDA template

---
 src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu b/src/pystencils_autodiff/backends/torch_native_cuda.tmpl.cu
index bd80ada..5edd28b 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 %}
-- 
GitLab