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