diff --git a/csrc/fused_dense_cuda.cu b/csrc/fused_dense_cuda.cu index 15c076f68..50299c90d 100644 --- a/csrc/fused_dense_cuda.cu +++ b/csrc/fused_dense_cuda.cu @@ -142,7 +142,7 @@ int gemm_lt( { hipStream_t stream; - hipblasHandle_t handle = at::cuda::getCurrentCUDABlasHandle(); + hipblasHandle_t handle = at::cuda::getCurrentCUDABlasLtHandle(); hipblasGetStream(handle, &stream); #if DEBUG