1#pragma once
2#include <c10/macros/Export.h>
3#include <c10/util/Flags.h>
4
5C10_DECLARE_bool(torch_lazy_ir_debug);
6C10_DECLARE_bool(torch_lazy_handle_special_scalars);
7C10_DECLARE_bool(torch_lazy_all_numbers_special_scalars);
8C10_DECLARE_bool(torch_lazy_param_aliasing);
9C10_DECLARE_bool(torch_lazy_reuse_ir);
10C10_DECLARE_bool(torch_lazy_use_thread_pool);
11
12C10_DECLARE_int(torch_lazy_compilation_cache_size);
13C10_DECLARE_int(torch_lazy_device_data_cache_size);
14C10_DECLARE_int(torch_lazy_io_thread_pool_size);
15C10_DECLARE_int(torch_lazy_metrics_samples);
16C10_DECLARE_int(torch_lazy_trim_graph_check_frequency);
17C10_DECLARE_int(torch_lazy_trim_graph_size);
18
19C10_DECLARE_string(torch_lazy_metrics_percentiles);
20
21C10_DECLARE_int(torch_lazy_shape_cache_size);
22
23namespace torch {
24namespace lazy {
25TORCH_API std::string& getLTCForceFallback();
26}
27} // namespace torch
28