pytorch/test/cpp_extensions
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
..
no_python_abi_suffix_test
self_compiler_include_dirs_test allow user passing relative paths in include_dirs within setuptools.setup (#38264) 2020-05-13 20:00:12 -07:00
torch_test_cpp_extension remediation of S205607 2020-07-17 17:19:47 -07:00
cpp_c10d_extension.cpp [torch distributed] Implementing all_gather_base (#56315) 2021-04-23 14:16:47 -07:00
cpp_c10d_extension.hpp [torch distributed] Implementing all_gather_base (#56315) 2021-04-23 14:16:47 -07:00
cpp_frontend_extension.cpp
cublas_extension.cpp Resubmit #67161 (#67735) 2021-11-04 09:59:30 -07:00
cuda_extension_kernel.cu Check CUDA kernel launches (/fbcode/caffe2/) (#49145) 2020-12-10 10:43:03 -08:00
cuda_extension_kernel2.cu Check CUDA kernel launches (/fbcode/caffe2/) (#49145) 2020-12-10 10:43:03 -08:00
cuda_extension.cpp
cuda_extension.cu Check CUDA kernel launches (/fbcode/caffe2/) (#49145) 2020-12-10 10:43:03 -08:00
cudnn_extension.cpp
cusolver_extension.cpp Resubmit #67161 (#67735) 2021-11-04 09:59:30 -07:00
dangling_impl_extension.cpp Add a test case for findDanglingImpls (#61104) 2021-07-07 13:34:16 -07:00
doubler.h
extension.cpp allow user passing relative paths in include_dirs within setuptools.setup (#38264) 2020-05-13 20:00:12 -07:00
jit_extension.cpp
jit_extension2.cpp
ort_extension.cpp Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
rng_extension.cpp Move generator state APIs to ATen (#49589) 2021-01-06 18:26:56 -08:00
setup.py [Reland] Add BUILD_LAZY_CUDA_LINALG option (#73447) 2022-03-01 21:33:07 +00:00
torch_library.cu Fix UB in library.h (#57962) 2021-05-11 16:04:02 -07:00