Skip to content
Snippets Groups Projects
Commit 02354656 authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

Fixed include of cuda.h in AtomicOperations.

parent 26ed46e9
No related branches found
No related tags found
1 merge request!48Segments
...@@ -12,7 +12,9 @@ ...@@ -12,7 +12,9 @@
#pragma once #pragma once
#ifdef HAVE_CUDA
#include <cuda.h> #include <cuda.h>
#endif
#include <TNL/Devices/Sequential.h> #include <TNL/Devices/Sequential.h>
#include <TNL/Devices/Host.h> #include <TNL/Devices/Host.h>
#include <TNL/Devices/Cuda.h> #include <TNL/Devices/Cuda.h>
...@@ -80,7 +82,7 @@ struct AtomicOperations< Devices::Cuda > ...@@ -80,7 +82,7 @@ struct AtomicOperations< Devices::Cuda >
TNL_ASSERT_TRUE( false, "Atomic add for long int is not supported on CUDA." ); TNL_ASSERT_TRUE( false, "Atomic add for long int is not supported on CUDA." );
#endif // HAVE_CUDA #endif // HAVE_CUDA
} }
__cuda_callable__ __cuda_callable__
static void add( short int& v, const short int& a ) static void add( short int& v, const short int& a )
{ {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment