diff options
Diffstat (limited to 'aten')
| -rw-r--r-- | aten/src/ATen/cuda/detail/CUDAHooks.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/aten/src/ATen/cuda/detail/CUDAHooks.cpp b/aten/src/ATen/cuda/detail/CUDAHooks.cpp index c93ef77222..eb9b2f32d1 100644 --- a/aten/src/ATen/cuda/detail/CUDAHooks.cpp +++ b/aten/src/ATen/cuda/detail/CUDAHooks.cpp @@ -33,6 +33,9 @@ std::unique_ptr<THCState, void (*)(THCState*)> CUDAHooks::initCUDA() const { THCState* thc_state = THCState_alloc(); THCudaInit(thc_state); +#ifdef USE_MAGMA + THCMagma_init(thc_state); +#endif return std::unique_ptr<THCState, void (*)(THCState*)>( thc_state, [](THCState* p) { if (p) |