pytorch/caffe2/core
Edward Yang b4a35632f9 Add function to materialize COW storages (#117053)
Summary: From Kurt Mohler, see https://github.com/pytorch/pytorch/pull/113396 (manually imported due to ghimport problems)

Test Plan: sandcastle, OSS CI

Differential Revision: D52610522

Pull Request resolved: https://github.com/pytorch/pytorch/pull/117053
Approved by: https://github.com/malfet, https://github.com/kurtamohler
2024-01-10 15:34:16 +00:00
..
hip Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
nomnigraph [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
__init__.py
allocator.cc
allocator.h
blob_gpu_test.cc
blob_serialization_gpu.cc
blob_serialization.cc [caffe2] Don't copy Tensor dims during deserialization (#79471) 2022-07-12 21:36:26 +00:00
blob_serialization.h [caffe2] Don't copy Tensor dims during deserialization (#79471) 2022-07-12 21:36:26 +00:00
blob_serializer_base.h [caffe2] add an EstimateAllBlobSizes operator (#59775) 2021-06-24 16:55:22 -07:00
blob_stats.cc
blob_stats.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
blob_test.cc Fix sign-compare in caffe2 cpp tests 2022-04-05 00:08:05 +00:00
blob.h [caffe2] Micro-optimizations in BlobGetMutableTensor (#98103) 2023-04-10 19:43:02 +00:00
CMakeLists.txt Remove caffe2 mobile (#84338) 2022-09-08 01:49:55 +00:00
common_cudnn.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
common_cudnn.h
common_gpu.cc [ROCm] use hipblas instead of rocblas (#105881) 2023-07-31 20:42:55 +00:00
common_gpu.h [ROCm] more 6.0 changes (#115946) 2023-12-20 20:19:29 +00:00
common_omp.h
common.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
common.h [caffe2] Avoid attempt to use undefined preprocessor directive 2021-06-05 09:22:52 -07:00
context_base.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
context_base.h External stream (#59527) 2021-06-14 13:46:11 -07:00
context_gpu_test.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
context_gpu.cu Add function to materialize COW storages (#117053) 2024-01-10 15:34:16 +00:00
context_gpu.h [caffe2] dont call cudnnDestroy on thread exit (crashes on windows with cuda 11/12) (#95382) 2023-03-10 06:42:51 +00:00
context_test.cc cleanup unused include (#93359) 2023-02-04 02:15:50 +00:00
context.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
context.h use irange for loops 2 (#66746) 2021-12-10 04:26:23 -08:00
cudnn_wrappers.h Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
db.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
db.h use irange for loops 2 (#66746) 2021-12-10 04:26:23 -08:00
distributions_stubs.h caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987) 2020-10-16 16:08:35 -07:00
event_cpu.h
event_gpu_test.cc
event_gpu.cc cast return of cudaGetLastError() to void when discarding (#62518) 2021-08-03 11:17:22 -07:00
event_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
event.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
event.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
export_c10_op_to_caffe2.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
export_c10_op_to_caffe2.h use irange for loops 2 (#66746) 2021-12-10 04:26:23 -08:00
export_caffe2_op_to_c10.h Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
flags.h
graph_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
graph.cc Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
graph.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
init_denormals.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
init_intrinsics_check.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
init_omp.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
init_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
init.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
init.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
int8_serialization.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
logging.h
macros.h
macros.h.in [EZ] Add CUSPARSELT to build variables (#116213) 2023-12-21 01:02:11 +00:00
memonger.cc [clang-tidy] Exclude cppcoreguidelines-avoid-magic-numbers (#57841) 2021-05-07 20:02:33 -07:00
memonger.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
module_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
module.cc
module.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
net_async_base.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net_async_base.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
net_async_scheduling.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net_async_scheduling.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
net_async_task_future.cc Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
net_async_task_future.h
net_async_task_graph.cc
net_async_task_graph.h
net_async_task.cc Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
net_async_task.h
net_async_tracing_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net_async_tracing.cc [Reland] Eliminate invocations of c10::stoi,c10::stod,c10::stoull,c10::stoll (#109566) 2023-09-19 07:15:25 +00:00
net_async_tracing.h Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
net_dag_utils_test.cc Fix warnings (#62930) 2021-08-11 14:07:10 -07:00
net_dag_utils.cc [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
net_dag_utils.h
net_gpu_test.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
net_parallel.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net_parallel.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
net_simple_refcount_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net_simple_refcount.cc [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
net_simple_refcount.h
net_simple.cc [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
net_simple.h [Caffe2] Implement BlackBoxPredictor::BenchmarkIndividualOps (#52903) 2021-02-27 19:49:22 -08:00
net_test.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
net.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
net.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
numa.cc
numa.h
observer_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
observer.h
operator_gpu_test.cc
operator_gradient.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
operator_schema_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
operator_schema.cc [PyTorch] Remove unnecessary iostream includes in headers (#61500) 2021-08-19 18:54:51 -07:00
operator_schema.h Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
operator_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
operator.cc [caffe2] Remove OperatorBase::newstyle_outputs_ (#67093) 2023-01-23 22:41:59 +00:00
operator.h [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
parallel_net_test.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
plan_executor_test.cc Fix broken caffe2 test: PlanExecutorTest.BlockingErrorPlan (#64401) 2021-09-02 08:30:29 -07:00
plan_executor.cc some reference and move fixes (#95942) 2023-03-10 03:44:09 +00:00
plan_executor.h
prof_dag_counters.cc turn on -Werror=type-limits in our Bazel CPU build 2022-06-10 10:04:08 +00:00
prof_dag_counters.h
qtensor_serialization.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
qtensor_serialization.h [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +00:00
qtensor.cc
qtensor.h Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
scope_guard.h
serialization_test.cc Fix sign-compare in caffe2 cpp tests 2022-04-05 00:08:05 +00:00
stats_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
stats.cc Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
stats.h [caffe2] Replace CAFFE_ prefixes in static_tracepoint.h macros with TORCH_ (#106380) 2023-08-03 21:51:36 +00:00
storage.h
tensor_impl.h
tensor_int8.cc
tensor_int8.h
tensor.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
tensor.h [PyTorch] Further reduce cost of TypeMeta::_typeMetaData (by 10x!) (#98105) 2023-04-12 17:44:48 +00:00
test_utils.cc [Caffe2][Testing] Check for equality first in assertTensorEqualsWithType<float> (#61006) 2021-06-29 23:31:37 -07:00
test_utils.h use irange for loops 2 (#66746) 2021-12-10 04:26:23 -08:00
timer_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
timer.h
transform_test.cc Fix sign-compare in caffe2 cpp tests 2022-04-05 00:08:05 +00:00
transform.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
transform.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
types.cc Speed up DataTypeToTypeMeta (#66113) 2021-10-07 08:06:09 -07:00
types.h Speed up DataTypeToTypeMeta (#66113) 2021-10-07 08:06:09 -07:00
workspace_test.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
workspace.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
workspace.h [caffe2] Avoid some double (and triple) lookups in workspace (#53319) 2021-03-04 22:57:02 -08:00