diff --git a/src/cuda/CUDAStream.cu b/src/cuda/CUDAStream.cu index b467d00..778a044 100644 --- a/src/cuda/CUDAStream.cu +++ b/src/cuda/CUDAStream.cu @@ -51,7 +51,7 @@ CUDAStream::CUDAStream(const int ARRAY_SIZE, const int device_index) // Check buffers fit on the device cudaDeviceProp props; cudaGetDeviceProperties(&props, 0); - if (props.totalGlobalMem < 3*ARRAY_SIZE*sizeof(T)) + if (props.totalGlobalMem < std::size_t{3}*ARRAY_SIZE*sizeof(T)) throw std::runtime_error("Device does not have enough memory for all 3 buffers"); // Create device buffers