pytorch/test/cpp_extensions
2024-03-08 20:01:03 +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
cuda_dlink_extension_add.cu
cuda_dlink_extension_add.cuh
cuda_dlink_extension_kernel.cu
cuda_dlink_extension.cpp
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cuda_extension.cpp
cuda_extension.cu
cudnn_extension.cpp
cusolver_extension.cpp
dangling_impl_extension.cpp
doubler.h
extension.cpp
identity.cpp [functorch] disable C++ Function under functorch transforms (#103957) 2023-06-23 11:01:44 +00:00
jit_extension.cpp
jit_extension2.cpp
mps_extension.mm [MPS] Prerequisite for MPS C++ extension (#102483) 2023-06-07 17:28:31 +00:00
open_registration_extension.cpp Modify StorageImplCreateHelper (#118459) 2024-03-07 06:26:55 +00:00
ort_extension.cpp
rng_extension.cpp Revert "Change ATEN generator argument type to const std::optional<Generator>& (#120076)" 2024-03-08 20:01:03 +00:00
setup.py [MPS] Prerequisite for MPS C++ extension (#102483) 2023-06-07 17:28:31 +00:00
torch_library.cu