reduce compiler warnings and avoid clash with KOKKOS

This commit is contained in:
Axel Kohlmeyer
2020-03-05 14:27:33 -05:00
parent b0c6641f1b
commit 5a3a5d86d4
2 changed files with 6 additions and 15 deletions

View File

@ -244,7 +244,7 @@ class UCL_Kernel {
template <class dtype>
inline void add_arg(const dtype* const arg) {
#if CUDA_VERSION >= 4000
_kernel_args[_num_args]=const_cast<dtype * const>(arg);
_kernel_args[_num_args]=const_cast<dtype *>(arg);
#else
_param_size = (_param_size+__alignof(dtype)-1) & ~(__alignof(dtype)-1);
CU_SAFE_CALL(cuParamSetv(_kernel,_param_size,(void*)arg,sizeof(dtype)));