pytorch
5b1f0ef6 - Add cuBLAS path for batched torch.geqrf (#56253)

Commit
4 years ago
Add cuBLAS path for batched torch.geqrf (#56253) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56253 `geqrfBatched` from cuBLAS is used if ``` (input.size(-2) <= 256 && batchCount(input) >= std::max<int64_t>(2, input.size(-2) / 16)) ``` Test Plan: Imported from OSS Reviewed By: ngimel Differential Revision: D27960156 Pulled By: mruberry fbshipit-source-id: 3e438eff01cbf7c7e075fb7aef709b97698a4650
Author
Parents
Loading