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