pytorch/test/cpp_extensions
Bug Hunter Yan 0c470b17e3 Extend storage create for custom storageImpl (#100237)
Fixes #ISSUE_NUMBER

For the scenario where users inherit storageimpl to implement their own subclasses, the current storage creation method cannot correctly create storage objects.

Refer to the registration method of Allocator to expand the creation method of storageimpl, users can register their own custom storageimpl creation.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100237
Approved by: https://github.com/albanD
2023-05-17 04:30:13 +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.cpp
cuda_dlink_extension_add.cu
cuda_dlink_extension_add.cuh
cuda_dlink_extension_kernel.cu
cuda_extension.cpp
cuda_extension.cu
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
cusolver_extension.cpp
dangling_impl_extension.cpp
doubler.h
extension.cpp Support large negative SymInt (#99157) 2023-04-15 22:43:51 +00:00
jit_extension.cpp
jit_extension2.cpp
open_registration_extension.cpp Extend storage create for custom storageImpl (#100237) 2023-05-17 04:30:13 +00:00
ort_extension.cpp
rng_extension.cpp Reland of "Python binding to set/get CUDA rng state offset" (#99565) 2023-04-20 15:42:25 +00:00
setup.py Fix shoud/shoudl typos (#97930) 2023-03-30 08:27:16 +00:00
torch_library.cu