| 234 | |
| 235 | template <class T> |
| 236 | static void CopyMemoryAsync(const T* from, T* to, ui64 size, const TCudaStream& stream) { |
| 237 | CUDA_SAFE_CALL(cudaMemcpyAsync(static_cast<void*>(to), static_cast<void*>(const_cast<T*>(from)), sizeof(T) * size, cudaMemcpyDefault, stream.GetStream())); |
| 238 | } |
| 239 | |
| 240 | inline void SetDevice(int devId) { |
| 241 | CUDA_SAFE_CALL(cudaSetDevice(devId)); |