
    קg5Q              	          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dZee         ed<   ej                            dd          dk    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(dZ)dZ* e+            Z,eee
                  ed<    e-ej                            dd	                    Z.ej                            dd          dk    Z/dZ0ej                            dd	          Z1ej                            dd	          Z2i Z3ee
e
f         ed<   h dZ4ej                            dd          Z5 e6ej                            dd                    Z7ej                            d          dk    Z8dZ9ej                            d          dk    Z:dZ;ej                            d          dk    Z<ej                            dd          dk    Z=d	Z>d	Z?d	Z@dZAdZBee-ef         ed<   ej                            dd          dk    ZCe-ed <   g d!ZDd" ZEdZFd	ZGdZHdZIdZJdZKd	ZLdZMdZNd	ZO e            rd	ndZPd	ZQdZR e e e eeS                                        ZTdZUd#ZVd$ZWd%ZXd	ZYej                            d&d          dk    ZZ e             Z[d	Z\d' Z] e]            Z^h d(Z_d	Z`d	Zad	ZbdZcdZddZe e+            Zfeee
gdf                  ed)<   d	Zgg d*Zhehi                    d+  ejj        ejk                  D                        ej                            d,d          dk    Zlej                            d-d          dk    Zmd	Znej                            d.d          dk    Zoerd d/lpT d0 Zqd d1lrmsZs  esejt        eu                    dS )2    N)abspathdirname)AnyCallableDictOptionalSetTypeTYPE_CHECKINGUnionc                  8    t          t          j        d           S )Ngit_version)hasattrtorchversion     P/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/_dynamo/config.py	is_fbcoder      s    u}m4444r   log_file_nameTORCHDYNAMO_VERBOSE01F   T      #TORCH_COMPILE_USE_LAZY_GRAPH_MODULEtraceable_tensor_subclassesTORCHDYNAMO_SUPPRESS_ERRORSTORCH_COMPILE_REPLAY_RECORD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?TORCHDYNAMO_REPRO_IGNORE_NON_FP"TORCHDYNAMO_CAPTURE_SCALAR_OUTPUTS,TORCHDYNAMO_CAPTURE_DYNAMIC_OUTPUT_SHAPE_OPSoptimize_ddp(TORCH_DYNAMO_DO_NOT_EMIT_RUNTIME_ASSERTSdo_not_emit_runtime_asserts)ddp_optimizerpython_reducer'python_reducer_without_compiled_forwardno_optimizationc                  :   t           j        t                   } t          | j        t
                    r| j        rd}nId}nFt          | j        t                    r| j        }n$t          dt          t                              || j	        v sJ d|            |S )Nr3   r6   z!Invalid type, type(optimize_ddp)=zInvalid mode mode=)
sysmodules__name__
isinstancer0   boolstr
ValueErrortype_ddp_optimization_mode)mmodes     r   _get_optimize_ddp_moderC   &  s    HA!.$'' A> 	%"DD$DD	ANC	(	( A~?$|*<*<??@@@1++++-DT-D-D+++Kr   float64
complex128int64TORCHDYNAMO_CPP_GUARD_MANAGERc                     d} | t           j        v r0t           j                            t           j        |          d          S t	                      rCt           j                            t          j                    t          j                    d          S t           j                            t          j	                    d          S )NTORCH_COMPILE_DEBUG_DIRtorch_compile_debug)
osenvironpathjoinr   tempfile
gettempdirgetpassgetusergetcwd)DEBUG_DIR_VAR_NAMEs    r   default_debug_dir_rootrU     s    2RZ''w||BJ'9:<QRRR	 @w||!!7?#4#46K
 
 	
 w||BIKK)>???r   >   repro_afterrepro_levelconstant_functionsr#   reorderable_logging_functions)striderequires_gradstorage_offsetlayoutdatac                 B    g | ]\  }}t          j        d |          |S )z^is_.*)rematch).0name_s      r   
<listcomp>re     s.    WWWgdARXiQU=V=VWTWWWr    TORCH_FAKE_TENSOR_DISPATCH_CACHE+TORCH_FAKE_TENSOR_DISPATCH_CACHE_CROSSCHECKTORCH_COMPILER_COLLECTIVES)*c                      d S )Nr   )changess    r   _make_closure_patcherrl     s    r   )install_config_module)vrQ   inspectrK   r`   r8   rO   os.pathr   r   typingr   r   r   r   r	   r
   r   r   r   r   r   r=   __annotations__rL   getverboseverify_correctnessminimum_call_countdead_code_eliminationcache_size_limitaccumulated_cache_size_limit&skip_code_recursive_on_cache_limit_hitspecialize_intspecialize_float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_ignorelistrV   intrW   repro_forward_onlyrepro_tolerancerepro_ignore_non_fpsame_two_models_use_fp64capture_scalar_outputs capture_dynamic_output_shape_ops+prefer_deferred_runtime_asserts_over_guards'allow_complex_guards_as_runtime_asserts3force_unspec_int_unbacked_size_like_on_torchrec_kjtenforce_cond_guards_matchr0   r2   r@   rC   skip_torchrecoptimize_ddp_lazy_compileskip_fsdp_guardsskip_fsdp_hooks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capture_sparse_computeerror_on_recompilereport_guard_failures__file__base_dirtrace_numpynumpy_default_floatnumpy_default_complexnumpy_default_intuse_numpy_random_streamenable_cpp_guard_managerinline_inbuilt_nn_modules%record_compile_time_instruction_countrU   debug_dir_root_save_config_ignore%cudagraph_backend_keep_input_mutation(cudagraph_backend_support_input_mutationonly_allow_pt2_compliant_opscapture_autograd_functioncapture_func_transformslog_compilation_metricsrY   +inject_BUILD_SET_unimplemented_TESTING_ONLY)_autograd_backward_strict_mode_banned_opsextend
getmembersTensorfake_tensor_cache_enabled$fake_tensor_cache_crosscheck_enabledcompiled_autogradenable_compiler_collectivestorch.utils._config_typingrl   torch.utils._config_modulerm   r9   r:   r   r   r   <module>r      s'     				 				 



  $ $ $ $ $ $ $ $ Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q 5 5 5  $x} # # # *...
4
4
;          #  *. & 
    JNN8#>>#E        !% 
 *. & "    $(  & /2cee Sc^ 3 3 3 $rz~~&CUKKLL
 
'DcJJcQ  $(   *..0%
8
8 :>>2E:: 57 !4S> 6 6 6. . . * jnn6== c"*..!:A>>?? Z^^$DEEL 
  jnn%FGG3N    
 (LMMQTT  JNNA3GG3N !
 /4 + +0 ' 7< 3 ! * "&eD#I % % % JNN=sCCsJ T        "  "    
 !  "   %  !     	 #,)++74 
     77777778#4#4556677    $      :>>*I3OOSVV  !*	O  ). %
@ 
@ 
@ ('))    ). % ,1 (  %        =@CEE s8SE4K#89 A A A /4 +- - - ) * 0 0WW++EL99WWW   JNN5s;;sB  JNN@#FF#M %   !jnn-I3OOSVV  ,,,,   = < < < < <  ck(+ , , , , ,r   