From 5a46ce238aab3892cb4b241790abb3cf5d879c15 Mon Sep 17 00:00:00 2001
From: Tomas Oberhuber <tomas.oberhuber@fjfi.cvut.cz>
Date: Thu, 20 Dec 2018 14:06:01 +0100
Subject: [PATCH] Fixing lambda function for CUDA in traverser benchmark.
---
src/Benchmarks/Traversers/WriteOne.h | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/Benchmarks/Traversers/WriteOne.h b/src/Benchmarks/Traversers/WriteOne.h
index 73bf0bfecf..9fd269f109 100644
--- a/src/Benchmarks/Traversers/WriteOne.h
+++ b/src/Benchmarks/Traversers/WriteOne.h
@@ -39,13 +39,13 @@ class WriteOne< 1, Device, Real, Index >
static void run( std::size_t size )
{
Vector v( size );
- auto writeOne = []( Index i, Real* data )
+ auto writeOne = [] __cuda_callable__ ( Index i, Real* data )
{
data[ i ] = 1.0;
};
- ParallelFor< Devices::Host >::exec( ( std::size_t ) 0, size, writeOne, v.getData() );
+ ParallelFor< Device >::exec( ( std::size_t ) 0, size, writeOne, v.getData() );
}
};
--
GitLab