Masaki Kozuki d4493b27ee make TORCH_(CUDABLAS|CUSOLVER)_CHECK usable in custom extensions (#67161) ... Summary: Make `TORCH_CUDABLAS_CHECK` and `TORCH_CUSOLVER_CHECK` available in custom extensions by exporting the internal functions called by the both macros. Rel: https://github.com/pytorch/pytorch/issues/67073 cc xwang233 ptrblck Pull Request resolved: https://github.com/pytorch/pytorch/pull/67161 Reviewed By: jbschlosser Differential Revision: D31984694 Pulled By: ngimel fbshipit-source-id: 0035ecd1398078cf7d3abc23aaefda57aaa31106		2021-10-29 17:27:07 -07: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	make TORCH_(CUDABLAS|CUSOLVER)_CHECK usable in custom extensions (#67161)	2021-10-29 17:27:07 -07:00
cuda_extension.cpp
cuda_extension.cu
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
cusolver_extension.cpp	make TORCH_(CUDABLAS|CUSOLVER)_CHECK usable in custom extensions (#67161)	2021-10-29 17:27:07 -07:00
dangling_impl_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
ort_extension.cpp
rng_extension.cpp
setup.py	make TORCH_(CUDABLAS|CUSOLVER)_CHECK usable in custom extensions (#67161)	2021-10-29 17:27:07 -07:00
torch_library.cu