pytorch/c10/core
Jessica Vandebon baff86dafb [MTIA tensor] allow shallow copy between CPU and MTIA tensors (#135871)
Reviewed By: egienvalue, hanzlfs

Differential Revision: D61662214

Pull Request resolved: https://github.com/pytorch/pytorch/pull/135871
Approved by: https://github.com/egienvalue, https://github.com/nautsimon
2024-09-13 22:13:58 +00:00
..
impl Update SavedTensorHooks TLS stack to use SafePyObject (#131700) 2024-08-02 16:27:16 +00:00
alignment.h
Allocator.cpp [CUDACachingAllocator] Turn Allocator::allocate into non-const (#120969) 2024-03-05 09:53:05 +00:00
Allocator.h [Memory Snapshot] Make recordAnnotations callback initialize lazily (#129242) 2024-06-22 04:05:55 +00:00
AutogradState.cpp
AutogradState.h
Backend.h [1/N] Change static functions in headers to inline (#127727) 2024-06-03 04:34:36 +00:00
BUILD.bazel
build.bzl Add function to materialize COW storages (#117053) 2024-01-10 15:34:16 +00:00
CachingDeviceAllocator.h [Reland] Refactor caching device allocator utils (#130923) 2024-09-07 11:14:17 +00:00
CompileTimeFunctionPointer.h
ConstantSymNodeImpl.cpp Add is_nested_int() (#119975) 2024-02-21 21:10:02 +00:00
ConstantSymNodeImpl.h [3/N] Fix Wunused-parameter warnings (#131271) 2024-07-20 23:31:03 +00:00
Contiguity.h Introduce size oblivious guards (#118579) 2024-02-06 19:45:32 +00:00
CopyBytes.cpp
CopyBytes.h
CPUAllocator.cpp [CUDACachingAllocator] Turn Allocator::allocate into non-const (#120969) 2024-03-05 09:53:05 +00:00
CPUAllocator.h
DefaultDtype.cpp
DefaultDtype.h
DefaultTensorOptions.h
Device.cpp rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
Device.h rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
DeviceArray.h
DeviceGuard.h [12/N] Use std::optional (#132361) 2024-08-02 13:46:46 +00:00
DeviceType.cpp [Fix] Check name when registering privateuse1 backend (#134071) 2024-08-27 20:28:30 +00:00
DeviceType.h rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
DispatchKey.cpp [MPS] Add support for autocast in MPS (#99272) 2024-09-05 23:23:17 +00:00
DispatchKey.h [MPS] Add support for autocast in MPS (#99272) 2024-09-05 23:23:17 +00:00
DispatchKeySet.cpp
DispatchKeySet.h [MPS] Add support for autocast in MPS (#99272) 2024-09-05 23:23:17 +00:00
DynamicCast.h Support unsigned int for randint, item, equality, fill, iinfo, tensor (#116805) 2024-01-10 02:17:23 +00:00
Event.h Build device generic torch.Stream and torch.Event based on c10::Stream/Event (#123611) 2024-04-24 20:51:17 +00:00
GeneratorImpl.cpp Graph-Safe RNG State Exchange for Tensor Parallelism (#114068) 2024-03-27 01:14:38 +00:00
GeneratorImpl.h Graph-Safe RNG State Exchange for Tensor Parallelism (#114068) 2024-03-27 01:14:38 +00:00
GradMode.cpp
GradMode.h
InferenceMode.cpp
InferenceMode.h
Layout.h Make SparseCsr a functionality dispatch key (#120703) 2024-03-01 13:28:46 +00:00
MemoryFormat.h
OptionalRef.h
PyHandleCache.h
QEngine.h
QScheme.h
RefcountedDeleter.cpp
RefcountedDeleter.h
SafePyObject.cpp
SafePyObject.h Update SavedTensorHooks TLS stack to use SafePyObject (#131700) 2024-08-02 16:27:16 +00:00
Scalar.cpp
Scalar.h [inductor] fix scalar miss constuctor for long type. (#132117) 2024-07-31 15:40:48 +00:00
ScalarType.cpp [custom_op] support default dtype values (#129189) 2024-06-23 00:13:23 +00:00
ScalarType.h [custom_op] support default dtype values (#129189) 2024-06-23 00:13:23 +00:00
ScalarTypeToTypeMeta.h [9/N] Replace c10::optional with std::optional (#130674) 2024-07-15 00:48:43 +00:00
Storage.cpp
Storage.h
StorageImpl.cpp Revert "Invalidate StorageImpl instances when tensor is overwritten with cudagraphs (#125264)" 2024-07-19 22:58:51 +00:00
StorageImpl.h Revert "Invalidate StorageImpl instances when tensor is overwritten with cudagraphs (#125264)" 2024-07-19 22:58:51 +00:00
Stream.cpp
Stream.h
StreamGuard.h [12/N] Use std::optional (#132361) 2024-08-02 13:46:46 +00:00
SymbolicShapeMeta.cpp [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
SymbolicShapeMeta.h Fix typo in assignment operators (#131890) 2024-07-27 11:13:42 +00:00
SymBool.cpp Remove unused variables (#122496) 2024-03-22 18:04:09 +00:00
SymBool.h [3/N] Fix Wunused-parameter warnings (#131271) 2024-07-20 23:31:03 +00:00
SymFloat.cpp
SymFloat.h
SymInt.cpp [BE] Use __builtin_overflow_sub when available (#117015) 2024-01-10 00:50:09 +00:00
SymInt.h [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
SymIntArrayRef.cpp
SymIntArrayRef.h [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
SymNodeImpl.cpp
SymNodeImpl.h C10_DIAGNOSTIC_PUSH_AND_IGNORED_IF_DEFINED and C10_DIAGNOST should be used in pairs (#135004) 2024-09-04 13:14:23 +00:00
TensorImpl.cpp Fix tensor.data access under inference_mode and compile (#134878) 2024-09-04 17:55:41 +00:00
TensorImpl.h [MTIA tensor] allow shallow copy between CPU and MTIA tensors (#135871) 2024-09-13 22:13:58 +00:00
TensorOptions.cpp
TensorOptions.h [9/N] Replace c10::optional with std::optional (#130674) 2024-07-15 00:48:43 +00:00
thread_pool.cpp [pytorch] Name threads in thread pools for better debugging (#130270) 2024-07-09 08:03:47 +00:00
thread_pool.h
UndefinedTensorImpl.cpp [inductor] fix UndefinedTensorImpl singleton can't export on Windows. (#132326) 2024-08-01 13:37:12 +00:00
UndefinedTensorImpl.h [inductor] fix UndefinedTensorImpl singleton can't export on Windows. (#132326) 2024-08-01 13:37:12 +00:00
WrapDimMinimal.cpp Enable nested namespace check in clang-tidy (#118506) 2024-01-31 00:32:35 +00:00
WrapDimMinimal.h