Provide source_generator::set_smem(size_t) for CUDA backend

ddemidov · Oct 20, 2016 · 12f0e32 · 12f0e32
1 parent faf15e4
commit 12f0e32
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/vexcl/backend/cuda/kernel.hpp b/vexcl/backend/cuda/kernel.hpp
@@ -113,6 +113,11 @@ class kernel {
             push_arg(arg.raw());
         }
 
+        /// Adds local memory to the kernel.
+        void set_smem(size_t smem_per_thread) {
+            smem = workgroup_size() * smem_per_thread;
+        }
+
         /// Adds local memory to the kernel.
         template <class F>
         void set_smem(F &&f) {