1 | #pragma once |
---|---|
2 | |
3 | #include <torch/csrc/Export.h> |
4 | #include <string> |
5 | |
6 | namespace torch { |
7 | namespace cuda { |
8 | |
9 | // C++-only versions of these, for python use |
10 | // those defined in cuda/Module.cpp which also record python state. |
11 | TORCH_CUDA_CU_API void _record_memory_history( |
12 | bool enabled, |
13 | int64_t alloc_trace_max_entries = 1); |
14 | TORCH_CUDA_CU_API std::string _memory_snapshot_pickled(); |
15 | |
16 | } // namespace cuda |
17 | } // namespace torch |
18 |