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