o
    Ih&                     @   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_moduleFFAKE_ALLOW_META10AOT_PARTITIONER_DEBUGT)	is_fbcodez0pytorch/remote_cache:enable_local_autograd_cacheTORCHINDUCTOR_AUTOGRAD_CACHE)justknob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 )N#TORCHINDUCTOR_AUTOGRAD_REMOTE_CACHEr   Tr   F)osenvironget r   r   K/var/www/vscode/kcb/lib/python3.10/site-packages/torch/_functorch/config.pyremote_autograd_cache_default2   s
   r   i  g      ?flopsdp PARTITIONER_MEMORY_BUDGET_PARETO$PARTITIONER_MEMORY_BUDGET_PARETO_DIRTORCH_COMPILE_GRAPH_FORMATsvg)*)4__doc__r   systypingr   r   torch.utils._config_moduler   r   functionalize_rng_opsr   r   fake_tensor_allow_metadebug_assertdebug_partitionerdecompose_custom_triton_opsstatic_weight_shapescsetorch._inductor.configr
   r   bool__annotations__r   enable_remote_autograd_cacheview_replay_for_aliased_outputsmax_dist_from_bwban_recompute_used_far_apart!ban_recompute_long_fusible_chains#ban_recompute_materialized_backwardban_recompute_not_in_allowlistban_recompute_reductionsrecompute_viewsactivation_memory_budget*activation_memory_budget_runtime_estimatoractivation_memory_budget_solvervisualize_memory_budget_paretomemory_budget_pareto_diraggressive_recomputation(fake_tensor_allow_unsafe_data_ptr_accessunlift_effect_tokensfake_tensor_crossref"fake_tensor_propagate_real_tensorsdonated_buffertorch_compile_graph_format*generate_fake_kernels_from_real_mismatchesgraphsafe_rng_functionalizationstrict_autograd_cache5disable_guess_zero_tangent_for_mutated_input_subclasstorch.utils._config_typingmodules__name__r   r   r   r   <module>   s`   