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 C10_DECLARE_bool(torch_lazy_enable_device_data_cache); 12 13 C10_DECLARE_int(torch_lazy_compilation_cache_size); 14 C10_DECLARE_int(torch_lazy_device_data_cache_size); 15 C10_DECLARE_int(torch_lazy_io_thread_pool_size); 16 C10_DECLARE_int(torch_lazy_metrics_samples); 17 C10_DECLARE_int(torch_lazy_trim_graph_check_frequency); 18 C10_DECLARE_int(torch_lazy_trim_graph_size); 19 20 C10_DECLARE_string(torch_lazy_metrics_percentiles); 21 22 C10_DECLARE_int(torch_lazy_shape_cache_size); 23 24 namespace torch { 25 namespace lazy { 26 TORCH_API std::string& getLTCForceFallback(); 27 } 28 } // namespace torch 29