pytorch/test/cpp_extensions
Youseok Yang b99e0c5412 Fix mtia_extension.cpp setDevice() to correctly set current_device (#149398)
We referred to this code and found that there was a minor bug. Fix for future reference for others.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/149398
Approved by: https://github.com/janeyx99
2025-03-31 06:07:22 +00:00
..
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
open_registration_extension OpenReg: fix issue of pin_memory (#145046) 2025-01-28 09:41:04 +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
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
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 [1/N] OpenReg: Replace open_registration_extension.cpp with openreg (#141815) 2025-01-14 15:59:00 +00:00
rng_extension.cpp
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