cub.h

Go to the documentation of this file.

#pragma once
 
#if defined(FZGMOD_BACKEND_HIP)
 
#include <hipcub/hipcub.hpp>
 
namespace cub = hipcub;
 
#elif defined(FZGMOD_BACKEND_SYCL)
 
#error "FZGMOD_BACKEND=SYCL is not implemented yet"
 
#else // FZGMOD_BACKEND_CUDA (default)
 
#include <cub/cub.cuh>
 
#endif