a
    hi5                     @   s  U d Z ddlZddlZddlmZ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
ZddlmZ eddd
dZeed< edddZeed< dZeed< ee d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-ed ed< dZ.dZ/e rNdnd
Z0ejd d!Z1dZ2d
Z3dZ4dZ5dZ6dZ7dZ8d"Z9erdd#l:T e	ej;e<  dS )$z
Global flags for aot autograd
    N)Literal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justknobenv_name_forcedefaultenable_autograd_cacheZ2TORCHINDUCTOR_AUTOGRAD_CACHE_ALLOW_CUSTOM_AUTOGRAD)r
   r   .autograd_cache_allow_custom_autograd_functionsbundled_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   E/var/www/auris/lib/python3.9/site-packages/torch/_functorch/config.pyremote_autograd_cache_default@   s
    r   i  g      ?ZflopsZdpZ PARTITIONER_MEMORY_BUDGET_PARETOZ$PARTITIONER_MEMORY_BUDGET_PARETO_DIRneeds_exact_strides)r   Zneeds_fixed_stride_orderZflexible_layout#custom_op_default_layout_constraintZTORCH_COMPILE_GRAPH_FORMATsvgZdonated)*)=__doc__r   systypingr   r   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 treat_parameters_as_free_to_saveZcseZtorch._environmentr	   r   bool__annotations__r   r   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_tokensr   Z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_cacheZ(unsafe_allow_optimization_of_collectivesZ5disable_guess_zero_tangent_for_mutated_input_subclassZ guess_tangent_strides_as_outputsZ_broadcast_rank0_decisionZ"saved_tensors_hooks_filtering_modeZtorch.utils._config_typingmodules__name__r   r   r   r   <module>   sx   