pytorch/test/cpp_extensions
Kurt Mohler 3706803b60 Change StorageImpl to track byte count rather than element count (#37776)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37776

* Remove type-specific size tracking in favor of byte size tracking in Storage and StorageImpl
* Changed numel() and set_numel() to nbytes() and set_nbytes()
* Added enum argument to Storage/StorageImpl constructor to indicate new meaning of the size parameter
* Update all callers of the changed API

Part of issue https://github.com/pytorch/pytorch/issues/33950
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37028

Differential Revision: D21171334

Pulled By: ezyang

fbshipit-source-id: 37329a379de9a3a83cc5e9007e455a3e1c2d10b8
2020-05-05 14:20:51 -07:00
..
no_python_abi_suffix_test
torch_test_cpp_extension
cpp_c10d_extension.cpp add c10d dynamic loading mechanism and unit test (#28068) 2020-04-02 15:46:51 -07:00
cpp_c10d_extension.hpp add c10d dynamic loading mechanism and unit test (#28068) 2020-04-02 15:46:51 -07:00
cpp_frontend_extension.cpp
cuda_extension.cpp
cuda_extension.cu
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
msnpu_extension.cpp Change StorageImpl to track byte count rather than element count (#37776) 2020-05-05 14:20:51 -07:00
rng_extension.cpp Prerequisites for CSPRNG (#36631) 2020-04-24 12:25:37 -07:00
setup.py pybind11 type_caster for at::Generator and custom RNG python test (#34774) 2020-03-22 10:57:35 -07:00