o ,hi5@sUdZddlZddlZddlmZmZmZddlmZm Z dZ ej dddkZ dZej d ddkZd Zd Zd Zd Zdd lmZed d d dZeed<edddZeed<dZeed<deefddZeZe ZdZd Zd Z d Z!d Z"d Z#dZ$dZ%dZ&dZ'ej dddkZ(ej dZ)dZ*d Z+dZ,dZ-eded<dZ.dZ/erdnd Z0ej d d!Z1dZ2d Z3dZ4dZ5dZ6dZ7dZ8d"Z9erdd#l:Te ej;e<dS)$z Global flags for aot autograd N)LiteralOptional TYPE_CHECKING)Configinstall_config_moduleFFAKE_ALLOW_META10AOT_PARTITIONER_DEBUGT) is_fbcodez0pytorch/remote_cache:enable_local_autograd_cacheTORCHINDUCTOR_AUTOGRAD_CACHE)justknobenv_name_forcedefaultenable_autograd_cache2TORCHINDUCTOR_AUTOGRAD_CACHE_ALLOW_CUSTOM_AUTOGRAD)rr.autograd_cache_allow_custom_autograd_functionsbundled_autograd_cachereturncCs,tjddkr dStjddkrdSdS)N#TORCHINDUCTOR_AUTOGRAD_REMOTE_CACHErTr F)osenvirongetrrR/var/www/html/scripts/venv/lib/python3.10/site-packages/torch/_functorch/config.pyremote_autograd_cache_default@s rig?flopsdp PARTITIONER_MEMORY_BUDGET_PARETO$PARTITIONER_MEMORY_BUDGET_PARETO_DIRneeds_exact_strides)r needs_fixed_stride_orderflexible_layout#custom_op_default_layout_constraintTORCH_COMPILE_GRAPH_FORMATsvgdonated)*)=__doc__rsystypingrrrtorch.utils._config_modulerrfunctionalize_rng_opsrrfake_tensor_allow_meta debug_assertdebug_partitionerdecompose_custom_triton_opsstatic_weight_shapes treat_parameters_as_free_to_savecsetorch._environmentr rbool__annotations__rrrenable_remote_autograd_cacheview_replay_for_aliased_outputsmax_dist_from_bwban_recompute_used_far_apart!ban_recompute_long_fusible_chains#ban_recompute_materialized_backwardban_recompute_not_in_allowlistban_recompute_reductionsrecompute_viewsactivation_memory_budget*activation_memory_budget_runtime_estimatoractivation_memory_budget_solvervisualize_memory_budget_paretomemory_budget_pareto_diraggressive_recomputation(fake_tensor_allow_unsafe_data_ptr_accessunlift_effect_tokensr#fake_tensor_crossref"fake_tensor_propagate_real_tensorsdonated_buffertorch_compile_graph_format*generate_fake_kernels_from_real_mismatchesgraphsafe_rng_functionalizationstrict_autograd_cache(unsafe_allow_optimization_of_collectives5disable_guess_zero_tangent_for_mutated_input_subclass guess_tangent_strides_as_outputs_broadcast_rank0_decision"saved_tensors_hooks_filtering_modetorch.utils._config_typingmodules__name__rrrrsz