pytorch/torch/backends
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
..
_coreml Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
_nnapi Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
cpu Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
cuda Allow setting TORCH_LINALG_PREFER_CUSOLVER=1 to prefer cusolver as linear algebra library globally (#106226) 2023-07-30 09:38:46 +00:00
cudnn Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
mkl Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
mkldnn Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
mps Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
openmp
opt_einsum Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
quantized Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
xeon Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
xnnpack Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00
__init__.py Apply UFMT to low traffic torch modules (#106249) 2023-07-29 23:37:30 +00:00