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	[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: 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	[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	[1/N] OpenReg: Replace open_registration_extension.cpp with openreg (#141815)	2025-01-14 15:59:00 +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