Shan19900305 8d12ba9acf add methods for open device in PackedSequence module. (#124923) ...

1) add is_{custom_device_name}() and {custom_device_name}() for open device register;
2) fix open device failed testcases.

@ezyang  @bdhirsh
Pull Request resolved: https://github.com/pytorch/pytorch/pull/124923
Approved by: https://github.com/ezyang

2024-04-26 15:26:20 +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_kernel2.cu

…

cuda_extension_kernel.cu

…

cuda_extension.cpp

…

cuda_extension.cu

…

cudnn_extension.cpp

…

cusolver_extension.cpp

…

dangling_impl_extension.cpp

…

doubler.h

…

extension.cpp

…

identity.cpp

…

jit_extension2.cpp

…

jit_extension.cpp

…

maia_extension.cpp

rename ort to maia (#123265)

2024-04-23 00:33:25 +00:00

mps_extension.mm

…

open_registration_extension.cpp

add methods for open device in PackedSequence module. (#124923)

2024-04-26 15:26:20 +00:00

rng_extension.cpp

Revert "Change ATEN generator argument type to const std::optional<Generator>& (#120076)"

2024-03-25 18:52:05 +00:00

setup.py

rename ort to maia (#123265)

2024-04-23 00:33:25 +00:00

torch_library.cu

…