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