
    i0=                        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                  j                  dd      dk7  ZdZej                  j                  d	d      dk7  Zd
Zd
Zd
Zd
ZddlmZ  eddd
      Zeed<    edd      Zeed<   dZeed<    e        Zdee   f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                  j                  dd      dk(  Z)ej                  j                  d      Z*dZ+d
Z,dZ-dZ.ed   ed<   dZ/dZ0dZ1 e       rdnd
Z2ej                  j                  d d!      Z3dZ4dZ5ee6   ed"<   d
Z7dZ8dZ9dZ:dZ;dZ<d#Z=erdd$l>  e	ej~                  e@          y)%z
Global flags for aot autograd
    N)Literal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_cache2TORCHINDUCTOR_AUTOGRAD_CACHE_ALLOW_CUSTOM_AUTOGRAD)r   r   .autograd_cache_allow_custom_autograd_functionsbundled_autograd_cachereturnc                      t         j                  j                  d      dk(  ryt         j                  j                  d      dk(  ryy )N#TORCHINDUCTOR_AUTOGRAD_REMOTE_CACHEr	   Tr
   F)osenvironget     Q/var/www/html/engine/venv/lib/python3.12/site-packages/torch/_functorch/config.pyremote_autograd_cache_defaultr   E   s7    	zz~~;<C	zz~~;<Cr   i  g      ?flopsdp PARTITIONER_MEMORY_BUDGET_PARETO$PARTITIONER_MEMORY_BUDGET_PARETO_DIRneeds_exact_strides)r#   needs_fixed_stride_orderflexible_layout#custom_op_default_layout_constraintsame_as_forwardTORCH_COMPILE_GRAPH_FORMATsvgfake_tensor_prefer_device_typedonated)*)A__doc__r   systypingr   r   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 treat_parameters_as_free_to_savecsetorch._environmentr   r   bool__annotations__r   r   autograd_cache_normalize_inputsr   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_tokensr&   fake_tensor_crossref"fake_tensor_propagate_real_tensorsbackward_pass_autocastdonated_buffertorch_compile_graph_format*generate_fake_kernels_from_real_mismatchesr*   strgraphsafe_rng_functionalizationstrict_autograd_cache(unsafe_allow_optimization_of_collectives5disable_guess_zero_tangent_for_mutated_input_subclass guess_tangent_strides_as_outputs_sync_decision_cross_ranks"saved_tensors_hooks_filtering_modetorch.utils._config_typingmodules__name__r   r   r   <module>r_      sX   
 
 3 3 D
   (93?3F  JJNN#:C@CG  #   $(    ( %?1 t  8>GQV8 .   %  $ '0k/ x~   => 0 '0k/   
  $  %) ! '+ #
 "&        .5 *
 #'  JJNN5s;sB  ::>>*PQ 
 ! 
 ,0 (    $WH&   < &+ "D +  $+4  ZZ^^,H%P 
 .3 * 15  4 #' 
  0 ,1 ( 9> 5 $)   #  &/ " , ckk(+ ,r   