1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#pragma once
#include <c10/macros/Export.h>
#include <string>
namespace torch::profiler::impl {
// Adds the execution trace observer as a global callback function, the data
// will be written to output file path.
TORCH_API bool addExecutionTraceObserver(const std::string& output_file_path);
// Remove the execution trace observer from the global callback functions.
TORCH_API void removeExecutionTraceObserver();
// Enables execution trace observer.
TORCH_API void enableExecutionTraceObserver();
// Disables execution trace observer.
TORCH_API void disableExecutionTraceObserver();
} // namespace torch::profiler::impl
|
