pytorch/torch/testing
Xiao Wang 21fd2bc32e Allow setting TORCH_LINALG_PREFER_CUSOLVER=1 to prefer cusolver as linear algebra library globally (#106226)
setting TORCH_LINALG_PREFER_CUSOLVER=1

This will allow users to prefer cusolver as linear algebra backend in their container use case. The switch is not enabled by default so it won't change any existing default behavior.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106226
Approved by: https://github.com/lezcano
2023-07-30 09:38:46 +00:00
..
_internal Allow setting TORCH_LINALG_PREFER_CUSOLVER=1 to prefer cusolver as linear algebra library globally (#106226) 2023-07-30 09:38:46 +00:00
__init__.py
_comparison.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
_creation.py disallow invalid value ranges in torch.testing.make_tensor (#96334) 2023-03-24 23:55:17 +00:00