pytorch/test
Nikita Shulga 6302cdb9bc [Reland] Add BUILD_LAZY_CUDA_LINALG option (#73447)
Summary:
When enabled, it will generate `torch_cuda_linalg` library, which would depend on cusolve and magma and registers dynamic bindings to it from LinearAlgebraStubs

Avoid symbol clashes that can result in infinite recursion by moving all symbols in the library to its own namespace.

Add checks that should prevent calling self in recursion to `LinearAlgebraStubs.cpp`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73447

Reviewed By: albanD

Differential Revision: D34538827

Pulled By: malfet

fbshipit-source-id: f2535b471d3524768a84b2e169b6aa24c26c03bf
(cherry picked from commit 4ec24b079c861c1122f0fa86e280b977c3c2f7ac)
2022-03-01 21:33:07 +00:00
..
ao/sparsity
autograd
benchmark_utils
bottleneck_test
cpp Allow custom registration of shape functions (#73270) 2022-02-28 17:44:45 +00:00
cpp_api_parity
cpp_extensions [Reland] Add BUILD_LAZY_CUDA_LINALG option (#73447) 2022-03-01 21:33:07 +00:00
custom_backend
custom_operator
distributed (torch/elastic) skip logging structured error info if error_file is not set (#73477) 2022-03-01 19:31:44 +00:00
distributions Add SoftplusTransform (#52300) 2022-02-25 02:30:03 +00:00
error_messages
expect Refactor FX codegen into extensible Codegen object (#72566) 2022-02-11 18:13:29 +00:00
forward_backward_compatibility Revert D34524207: [pytorch][PR] remove _s_where 2022-03-01 17:43:16 +00:00
fx [test_fx_const_fold] Remove dependencies on acc_* (#72810) 2022-02-14 20:46:54 +00:00
jit Support dataclasses in TorchScript (#73066) 2022-02-28 19:34:20 +00:00
jit_hooks
mobile Add dynamic shape support to AOT driver & compiler (#72995) 2022-02-24 04:30:48 +00:00
onnx Improve logging on ONNX symbolic helpers 2022-03-01 19:35:09 +00:00
package [torch.package] add test case for repackaging parent module (#72367) 2022-02-07 21:49:36 +00:00
quantization [quant][fx] Remove Fuser class in fusion implementation (#73470) 2022-03-01 09:29:21 +00:00
scripts
test_img
typing
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
run_test.py add dry run option and improve test list printing 2022-02-22 20:45:41 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py
test_autograd.py no longer coalesce sparse COO tensors before comparison (#69751) 2022-02-17 02:33:08 +00:00
test_binary_ufuncs.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_bundled_images.py
test_bundled_inputs.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py
test_cuda.py [c10][rocm] fix __assert_fail() declaration mismatch error (#73040) 2022-03-01 04:35:30 +00:00
test_cuda_primary_ctx.py remove hasPrimaryContext workaround on ROCm (#71146) 2022-01-25 20:32:12 +00:00
test_dataloader.py [DataPipe] Fix issue with DataPipe serialization with dill (#72896) 2022-02-23 16:31:20 +00:00
test_datapipe.py [DataPipe] Adding serialization test at different stages of reading for IterDataPipes (#73119) 2022-02-23 16:31:21 +00:00
test_deploy.py
test_determination.py
test_dispatch.py Revert D34034848: free up dispatch key space (in C++) 2022-02-14 23:29:00 +00:00
test_expanded_weights.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_foreach.py use torch.testing.assert_equal in TestCase.assertEqual (#67796) 2022-01-27 08:33:55 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py [Opt Overlap] Clean up code in _OptimizerHookState (#71620) 2022-01-26 19:33:49 +00:00
test_functionalization.py
test_futures.py
test_fx.py [fx] Improve support for tuple subclasses such as NamedTuple (#73198) 2022-02-23 11:31:10 +00:00
test_fx_experimental.py Remove some unnecessary python functional wrappers (#61608) 2022-02-01 16:59:26 +00:00
test_import_stats.py
test_indexing.py
test_jit.py Support dataclasses in TorchScript (#73066) 2022-02-28 19:34:20 +00:00
test_jit_autocast.py
test_jit_cuda_fuser.py Nvfuser code bump 2_1_2022 (#72127) 2022-02-15 00:43:16 +00:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Implement Tanh Gelu Approximation (#61439) 2022-02-14 03:40:32 +00:00
test_jit_legacy.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py
test_license.py
test_linalg.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_logging.py
test_masked.py Testing of masked reductions on mixed layout inputs. (#72398) 2022-02-25 05:32:47 +00:00
test_metal.py
test_mkldnn.py
test_mobile_optimizer.py Prepare for an update to the XNNPACK submodule (#72642) 2022-02-25 00:39:15 +00:00
test_model_dump.py Prepare for an update to the XNNPACK submodule (#72642) 2022-02-25 00:39:15 +00:00
test_module_init.py
test_modules.py only compare attributes for meta tensors (#72508) 2022-02-17 02:33:08 +00:00
test_monitor.py torch/monitor: merge Interval and FixedCount stats (#72009) 2022-01-30 23:21:59 +00:00
test_multiprocessing.py Allow forking until a worker thread is created in autograd engine (#72689) 2022-02-12 01:52:57 +00:00
test_multiprocessing_spawn.py Make test_multiprocessing_spawn.py compatible with pytest (#50408) 2022-02-10 16:18:46 +00:00
test_namedtensor.py use torch.testing.assert_equal in TestCase.assertEqual (#67796) 2022-01-27 08:33:55 +00:00
test_namedtuple_return_api.py Rewrite svd and linalg.svd as structured kernels (#69827) 2022-01-27 18:38:30 +00:00
test_native_functions.py
test_nn.py Optimize grid sample 3d 2022-02-23 19:25:17 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py Skip superfluous storage allocations while constructing meta tensors (#65331) 2022-02-11 12:55:11 +00:00
test_openmp.py
test_ops.py [ROCm] Enable skipped ROCm unit tests (#67706) 2022-02-11 22:14:54 +00:00
test_optim.py Uncomment step no-op test in test_optim (#70953) 2022-02-15 18:02:08 +00:00
test_overrides.py Add autograd tests for addmm, addmv, mm, mv and CSR matrix input (#71949) 2022-02-11 23:14:02 +00:00
test_package.py
test_per_overload_api.py Add torch.ops per overload API (#72206) 2022-02-11 17:19:06 +00:00
test_profiler.py Disable forward/backward correlation to workaround the profiler crash (#72904) 2022-03-01 20:34:14 +00:00
test_pruning_op.py
test_public_bindings.py
test_python_dispatch.py Fix serialization and deepcopying for wrapper subclasses 2022-02-24 18:21:25 +00:00
test_pytree.py
test_quantization.py
test_reductions.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_scatter_gather_ops.py
test_segment_reductions.py
test_serialization.py Fix serialization and deepcopying for wrapper subclasses 2022-02-24 18:21:25 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_show_pickle.py
test_sort_and_select.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_sparse.py Enable test_coalesce_cuda_bfloat16 (#73158) 2022-02-28 19:34:20 +00:00
test_sparse_csr.py Revert D33767740: [pytorch][PR] Sparse CSR CPU: cuSolverSP backend for linalg.solve 2022-03-01 18:33:23 +00:00
test_spectral_ops.py fft: Fix invalid shape error for complex-to-real transforms (#73012) 2022-02-23 03:33:26 +00:00
test_stateless.py Add lightweight reparametrization for _stateless calls (#68969) 2022-01-28 14:38:45 +00:00
test_static_runtime.py Fix quicklints (#72256) 2022-02-03 15:17:35 +00:00
test_tensor_creation_ops.py Refactor set_default_tensor_type to avoid legacy tensor types 2022-02-28 15:08:17 +00:00
test_tensorboard.py
test_tensorexpr.py [JIT] Opinfo tests for nnc fusion - retry (#72486) 2022-02-09 19:01:22 +00:00
test_tensorexpr_pybind.py [TensorExpr] Delet DimArg class. (#72390) 2022-02-11 01:21:59 +00:00
test_testing.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_throughput_benchmark.py
test_torch.py Revert D34524207: [pytorch][PR] remove _s_where 2022-03-01 17:43:16 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py only compare attributes for meta tensors (#72508) 2022-02-17 02:33:08 +00:00
test_typing.py
test_unary_ufuncs.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_utils.py
test_view_ops.py align signature of make_tensor with other creation ops (#72702) 2022-02-25 06:30:31 +00:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py Update XNNPACK (#73467) 2022-02-28 06:09:26 +00:00