o
    Zh&                     @   sT  U d Z ddlZddlZddlmZmZ ddlmZmZ dZ	ej
dddkZdZej
d	ddkZd
Zd
Zd
ZddlmZ eddd
dZeed< dee fddZe Ze  ZdZd
Zd
Zd
Zd
Zd
ZdZ dZ!dZ"dZ#ej
dddkZ$ej
dZ%dZ&d
Z'dZ(dZ)dZ*e rdnd
Z+ej
ddZ,dZ-d
Z.dZ/dZ0erddl1T eej2e3  dS )z
Global flags for aot autograd
    N)OptionalTYPE_CHECKING)Configinstall_config_moduleFZFAKE_ALLOW_META10ZAOT_PARTITIONER_DEBUGT)	is_fbcodez0pytorch/remote_cache:enable_local_autograd_cacheZTORCHINDUCTOR_AUTOGRAD_CACHE)ZjustknobZenv_name_forcedefaultenable_autograd_cachereturnc                   C   s,   t jddkr
dS t jddkrdS d S )NZ#TORCHINDUCTOR_AUTOGRAD_REMOTE_CACHEr   Tr   F)osenvironget r   r   F/var/www/auris/lib/python3.10/site-packages/torch/_functorch/config.pyremote_autograd_cache_default2   s
   r   i  g      ?ZflopsZdpZ PARTITIONER_MEMORY_BUDGET_PARETOZ$PARTITIONER_MEMORY_BUDGET_PARETO_DIRZTORCH_COMPILE_GRAPH_FORMATsvg)*)4__doc__r   systypingr   r   Ztorch.utils._config_moduler   r   Zfunctionalize_rng_opsr   r   Zfake_tensor_allow_metaZdebug_assertZdebug_partitionerZdecompose_custom_triton_opsZstatic_weight_shapesZcseZtorch._inductor.configr   r
   bool__annotations__r   Zenable_remote_autograd_cacheZview_replay_for_aliased_outputsZmax_dist_from_bwZban_recompute_used_far_apartZ!ban_recompute_long_fusible_chainsZ#ban_recompute_materialized_backwardZban_recompute_not_in_allowlistZban_recompute_reductionsZrecompute_viewsZactivation_memory_budgetZ*activation_memory_budget_runtime_estimatorZactivation_memory_budget_solverZvisualize_memory_budget_paretoZmemory_budget_pareto_dirZaggressive_recomputationZ(fake_tensor_allow_unsafe_data_ptr_accessZunlift_effect_tokensZfake_tensor_crossrefZ"fake_tensor_propagate_real_tensorsZdonated_bufferZtorch_compile_graph_formatZ*generate_fake_kernels_from_real_mismatchesZgraphsafe_rng_functionalizationZstrict_autograd_cacheZ5disable_guess_zero_tangent_for_mutated_input_subclassZtorch.utils._config_typingmodules__name__r   r   r   r   <module>   s`   