pytorch/torch/testing/_internal
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
..
codegen [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
data
distributed [DDP] Support optim in backward after DDP init (#105995) 2023-07-29 01:36:25 +00:00
generated
opinfo [foreach] Set SavedVariable.is_output to true for grad_fn->result_ (#105504) 2023-07-26 14:29:32 +00:00
optests [operator_compile_check] Add FakeTensor testing (#103595) 2023-06-16 16:55:51 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py [autograd_function_db] Add NumpyTake as OpInfo (#98438) 2023-04-10 18:04:50 +00:00
check_kernel_launches.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_cuda.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_device_type.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
common_dist_composable.py
common_distributed.py [Composable FSDP] Enable HSDP (#105206) 2023-07-26 21:03:55 +00:00
common_dtype.py
common_fsdp.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_jit.py
common_methods_invocations.py [pt2] add meta for argsort.stable, use sort samples in OpInfo (#106025) 2023-07-27 03:49:17 +00:00
common_modules.py Added ModuleInfo test for meta device ctx init (#105871) 2023-07-26 01:57:54 +00:00
common_nn.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
common_pruning.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_quantization.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
common_quantized.py
common_subclass.py
common_utils.py Allow setting TORCH_LINALG_PREFER_CUSOLVER=1 to prefer cusolver as linear algebra library globally (#106226) 2023-07-30 09:38:46 +00:00
composite_compliance.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
control_flow_opinfo_db.py Support map autograd and pytree in/out. (#101633) 2023-05-17 16:52:26 +00:00
custom_op_db.py Update custom op API (#105947) 2023-07-28 13:30:58 +00:00
dist_utils.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
hypothesis_utils.py
inductor_utils.py Enable default workflow PyT 2.0 UTs on ROCm stack (#100981) 2023-05-15 23:45:04 +00:00
jit_metaprogramming_utils.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
jit_utils.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
logging_tensor.py Upgrade LoggingTensor mode and add traceback collection (#103734) 2023-06-21 22:04:30 +00:00
logging_utils.py [Dynamo] Enable torch._dynamo.config.suppress_errors by default (#105307) 2023-07-21 19:17:46 +00:00
quantization_torch_package_models.py