Update on "Add cuBLAS path for batched torch.geqrf"

IvanYashchuk · IvanYashchuk · commit 7d029cfecbdc · 2021-04-26T10:32:25.000Z
`geqrfBatched` from cuBLAS is used if ``` (input.size(-2) <= 256 && batchCount(input) >= std::max<int64_t>(2, input.size(-2) / 16)) ``` Differential Revision: [D27960156](https://our.internmc.facebook.com/intern/diff/D27960156) [ghstack-poisoned]
diff --git a/aten/src/ATen/cuda/CUDABlas.h b/aten/src/ATen/cuda/CUDABlas.h
@@ -206,7 +206,7 @@ void vdot<c10::complex<double>>(CUDABLAS_DOT_ARGTYPES(c10::complex<double>));
 
 template <class Dtype>
 void geqrfBatched(CUDABLAS_GEQRF_BATCHED_ARGTYPES(Dtype)) {
-  TORCH_CHECK(
+  TORCH_INTERNAL_ASSERT(
       false,
       "at::cuda::blas::geqrfBatched: not implemented for ",
       typeid(Dtype).name());
