Shan19900305 6662627c89 Add APIs for custom device using TensorIteratorBase. (#120792) ...

1) add operand and get_dim_names API;
2) set will_resize to true when output tensor is undefined;
3) add abs_stub for dummy device and calculate on cpu device;
4) support dummy device copy with stride;
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120792
Approved by: https://github.com/ezyang

2024-03-20 03:51:09 +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

…

mps_extension.mm

…

open_registration_extension.cpp

Add APIs for custom device using TensorIteratorBase. (#120792)

2024-03-20 03:51:09 +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

…

torch_library.cu

…