Skip to content
Snippets Groups Projects
Commit ccdd64f9 authored by Pietro Incardona's avatar Pietro Incardona
Browse files

Fixing CUDIFY

parent f95b31b6
No related branches found
No related tags found
No related merge requests found
Pipeline #4052 passed
......@@ -8,6 +8,9 @@ constexpr int default_kernel_wg_threads_ = 1024;
#if CUDART_VERSION >= 11000 && defined(__NVCC__)
#include "cub/util_type.cuh"
#include "cub/block/block_scan.cuh"
#endif
#ifdef __NVCC__
template<typename lambda_f>
__global__ void kernel_launch_lambda(lambda_f f)
......
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