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