| .. |
|
libtorch_agnostic_extension
|
[BE] Add STABLE_LIBRARY test for multiple returns (#149230)
|
2025-03-18 02:40:54 +00:00 |
|
no_python_abi_suffix_test
|
[BE][Easy][7/19] enforce style for empty lines in import segments in test/[a-c]*/ and test/[q-z]*/ (#129758)
|
2024-07-31 10:54:03 +00:00 |
|
open_registration_extension
|
[OpenReg] Move SDPA to OpenReg from open_registration_extension.cpp (#153309)
|
2025-05-13 03:49:19 +00:00 |
|
python_agnostic_extension
|
Set -DPy_LIMITED_API flag for py_limited_api=True extensions (#145764)
|
2025-01-28 20:11:05 +00:00 |
|
self_compiler_include_dirs_test
|
|
|
|
torch_test_cpp_extension
|
[RFC] Add support for device extension autoloading (#127074)
|
2024-07-09 06:14:13 +00:00 |
|
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
|
Support torch.dtype as parameter in pybind11 cpp extension. (#126865)
|
2024-05-29 23:19:32 +00:00 |
|
identity.cpp
|
|
|
|
jit_extension.cpp
|
|
|
|
jit_extension2.cpp
|
|
|
|
maia_extension.cpp
|
[MAIA] [Autocast] Enable autocast on MAIA device (#148511)
|
2025-03-18 03:46:22 +00:00 |
|
mps_extension.mm
|
[BE] Use helper functions in mps_extension (#137313)
|
2024-10-04 00:26:38 +00:00 |
|
mtia_extension.cpp
|
Fix mtia_extension.cpp setDevice() to correctly set current_device (#149398)
|
2025-03-31 06:07:22 +00:00 |
|
open_registration_extension.cpp
|
[OpenReg] Move SDPA to OpenReg from open_registration_extension.cpp (#153309)
|
2025-05-13 03:49:19 +00:00 |
|
rng_extension.cpp
|
[codemod] c10:optional -> std::optional (#126135)
|
2024-05-14 19:35:51 +00:00 |
|
setup.py
|
xpu: support sycl with torch.utils.cpp_extension APIs (#132945)
|
2025-02-16 16:50:59 +00:00 |
|
torch_library.cu
|
|
|
|
xpu_extension.sycl
|
xpu: support sycl with torch.utils.cpp_extension APIs (#132945)
|
2025-02-16 16:50:59 +00:00 |
