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
torch_test_cpp_extension
cpp_c10d_extension.cpp
cpp_c10d_extension.hpp
cpp_frontend_extension.cpp
cublas_extension.cpp Resubmit #67161 (#67735) 2021-11-04 09:59:30 -07:00
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cuda_extension.cpp
cuda_extension.cu
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
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
setup.py [Reland] Add BUILD_LAZY_CUDA_LINALG option (#73447) 2022-03-01 21:33:07 +00:00
torch_library.cu