o
    ߗi$                     @   s<  d Z ddlZddl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ejd
ddkZdee fddZe ZddlmZ e  oMe 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dZ d	Z!dZ"dZ#dZ$e rzdnd	Z%ejddZ&dZ'dZ(dZ)erddl*T ddl+m,Z, e,ej-e.  dS )z
Global flags for aot autograd
    N)OptionalTYPE_CHECKINGFFAKE_ALLOW_META10AOT_PARTITIONER_DEBUGTTORCHINDUCTOR_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   U/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_functorch/config.pyremote_autograd_cache_default'   s
   r   )	is_fbcodei  g      ?flopsdp PARTITIONER_MEMORY_BUDGET_PARETOTORCH_COMPILE_GRAPH_FORMATsvg)*)install_config_module)/__doc__r   systypingr   r   functionalize_rng_opsr   r   fake_tensor_allow_metadebug_assertdebug_partitionerstatic_weight_shapescseenable_autograd_cacheboolr   enable_remote_autograd_cachetorch._inductor.configr   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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strict_autograd_cache5disable_guess_zero_tangent_for_mutated_input_subclasstorch.utils._config_typingtorch.utils._config_moduler   modules__name__r   r   r   r   <module>   sR   