
    '}hE              
          U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl	m
Z
mZmZmZmZmZmZmZ d dlZdZee   ed<   ej,                  j/                  dd      dk(  ZdZd	Zd
ZdZdZdZd
Zej,                  j/                  dd      dk(  Z d
Z!d
Z"d
Z#d
Z$dZ%dZ&d
Z' e(       Z)eee
      ed<    e*ej,                  j/                  dd            Z+ej,                  j/                  dd      dk(  Z,d
Z-ej,                  j/                  dd      Z.ej,                  j/                  dd      Z/i Z0ee
e
f   ed<   h dZ1ej,                  j/                  dd      Z2 e3ej,                  j/                  dd            Z4ej,                  j/                  d      dk(  Z5dZ6d
Z7dZ8dZ9dZ:d
Z;d
Z<ee*ef   ed<   g dZ=d Z>dZ?d
Z@d
ZAd
ZBdZCd
ZDd
ZEdZFdZGd
ZH e e e eeI                        ZJd
ZKdZLdZMdZNdZOdZPd  ZQd! ZR eR       ZSh d"ZTdZUdZVd
ZWdZXd
ZYd
ZZ e(       Z[eee
gdf      ed#<   dZ\g d$Z]e]j                   ej                  ej                        D  cg c]  \  } } ej                  d%|       s|  c}}        ej,                  j/                  d&d      dk(  Zbej,                  j/                  d'd      dk(  ZcdZderd d(le d) Zfd d*lgmhZh  ehej                  ej          yc c}} w )+    N)abspathdirname)AnyCallableDictOptionalSetTypeTYPE_CHECKINGUnionlog_file_nameTORCHDYNAMO_VERBOSE01F   T   @   #TORCH_COMPILE_USE_LAZY_GRAPH_MODULEtraceable_tensor_subclassesTORCHDYNAMO_SUPPRESS_ERRORSTORCH_COMPILE_DEBUGTORCH_COMPILE_DISABLETORCH_COMPILE_CPROFILE!skipfiles_inline_module_allowlist>   torch._refstorch._primstorch._decomptorch.testingtorch.distributionsTORCHDYNAMO_REPRO_AFTERTORCHDYNAMO_REPRO_LEVEL   TORCHDYNAMO_REPRO_FORWARD_ONLYgMbP?optimize_ddp)ddp_optimizerpython_reducer'python_reducer_without_compiled_forwardno_optimizationc                  @   t         j                  t           } t        | j                  t
              r| j                  rd}nEd}nBt        | j                  t              r| j                  }nt        dt        t                    || j                  v s
J d|       |S )Nr%   r(   z!Invalid type, type(optimize_ddp)=zInvalid mode mode=)
sysmodules__name__
isinstancer$   boolstr
ValueErrortype_ddp_optimization_mode)mmodes     S/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/_dynamo/config.py_get_optimize_ddp_moder6      s    HA!..$'>>"D$D	ANNC	(~~=$|*<)>?@@1+++D/BTG-DD+K    float64
complex128int64c                  8    t        t        j                  d       S )Ngit_version)hasattrtorchversion r7   r5   	is_fbcoderA   F  s    u}}m444r7   c                     d} | t         j                  v r1t         j                  j                  t         j                  |    d      S t	               rEt         j                  j                  t        j                         t        j                         d      S t         j                  j                  t        j                         d      S )NTORCH_COMPILE_DEBUG_DIRtorch_compile_debug)
osenvironpathjoinrA   tempfile
gettempdirgetpassgetusergetcwd)DEBUG_DIR_VAR_NAMEs    r5   default_debug_dir_rootrO   J  s    2RZZ'ww||BJJ'9:<QRR	ww||!7??#46K
 	
 ww||BIIK)>??r7   >   repro_afterrepro_levelconstant_functionsr   reorderable_logging_functions)striderequires_gradstorage_offsetlayoutdataz^is_.* TORCH_FAKE_TENSOR_DISPATCH_CACHE+TORCH_FAKE_TENSOR_DISPATCH_CACHE_CROSSCHECK)*c                       y )Nr@   )changess    r5   _make_closure_patcherr^     s    r7   )install_config_module)krK   inspectrE   rer*   rI   os.pathr   r   typingr   r   r   r   r	   r
   r   r   r>   r   r/   __annotations__rF   getverboseverify_correctnessminimum_call_countdead_code_eliminationcache_size_limitaccumulated_cache_size_limitspecialize_intdynamic_shapesuse_lazy_graph_moduleassume_static_by_defaultautomatic_dynamic_shapesforce_parameter_static_shapes&force_nn_module_property_static_shapesallow_ignore_mark_dynamicguard_nn_modules guard_nn_modules_using_dict_tagssetr   r.   suppress_errorsreplay_record_enabled rewrite_assert_with_torch_assertdisablecprofiler   *allowed_functions_module_string_ignorelistrP   intrQ   repro_forward_onlyrepro_tolerancesame_two_models_use_fp64capture_scalar_outputs capture_dynamic_output_shape_ops3force_unspec_int_unbacked_size_like_on_torchrec_kjtenforce_cond_guards_matchr$   r2   r6   optimize_ddp_lazy_compileskip_fsdp_guardsskip_nnmodule_hook_guardsraise_on_ctx_manager_usageraise_on_unsafe_aot_autograderror_on_nested_jit_traceerror_on_nested_fx_trace	allow_rnnerror_on_recompilereport_guard_failures__file__base_dirtrace_numpytrace_distributednumpy_default_floatnumpy_default_complexnumpy_default_intuse_numpy_random_streamrA   rO   debug_dir_root_save_config_ignore%cudagraph_backend_keep_input_mutationonly_allow_pt2_compliant_opscapture_autograd_functioncapture_func_transforms$optimize_user_defined_triton_kernelslog_compilation_metricsrS   +inject_BUILD_SET_unimplemented_TESTING_ONLY)_autograd_backward_strict_mode_banned_opsextend
getmembersTensormatchfake_tensor_cache_enabled$fake_tensor_cache_crosscheck_enabled:_experimental_support_context_fn_in_torch_utils_checkpointtorch.utils._config_typingr^   torch.utils._config_moduler_   r+   r,   )name_s   00r5   <module>r      s     	 	 
  $ Q Q Q   $x} # **...
4
;          "    JJNN8#>#E        !% 
 *. & "    $(  $ /2e Sc^ 3 rzz~~&CUKL
 

'<cBcI  $(   **..0%
8 ::>>2E: 57 !4S> 6. * jjnn6= "**..!:A>? ZZ^^$DEL 
    
   $)   7< 3 ! * "&eD#I % ( "   
 !  "   %  !     	
     7778#4567      $     5
@ ()  ). %  %       (, $   =@E s8SE4K#89 A /4 +- ) * 0 0+++ELL9WgdAXRXXiQU=VTW JJNN5s;sB  JJNN@#F#M % >C :, = ckk(+ ,5 Xs   K 
.K 
