o
    "i\N                     @   sp  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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!ejdd	d	kZ"dZ#dZ$dZ%dZ&d
Z'e rvd
n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Z/ejdd
Z0ejdd
Z1i Z2ee
e
f ed< h dZ3ejddZ4e5ejddZ6ejdd	kZ7dZ8ejdd	kZ9dZ:ejdd	kZ;ejddd	kZ<d
Z=d
Z>d
Z?dZ@dZAee,ef ed< ejd dd	kZBe,ed!< g d"ZCd#d$ ZDd
ZEdZFd
ZGdZHdZId
ZJdZKdZLd
ZMd
ZNdZOeeeeePZQdZRd
ZSd%ZTd&ZUd'ZVd
ZWejd(d	d	kZXejd)dd	kZYd*d+ ZZeZ Z[h d,Z\d
Z]d
Z^d
Z_dZ`dZadZbe* Zceee
gdf  ed-< d
Zdg d.Zeeefd/d0 egejhD  ejd1d	d	kZiejd2dd	kZjd
Zkd
Zlerd d3lmT d4d5 Znd d6lompZp epejqer  dS )7    N)abspathdirname)AnyCallableDictOptionalSetTypeTYPE_CHECKINGUnionc                   C   s   t tjd S )Ngit_version)hasattrtorchversion r   r   R/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/_dynamo/config.py	is_fbcode   s   r   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                  C   sj   t jt } t| jtr| jrd}nd}nt| jtr| j}n	tdtt|| j	v s3J d||S )Nr0   r3   z!Invalid type, type(optimize_ddp)=zInvalid mode mode=)
sysmodules__name__
isinstancer-   boolstr
ValueErrortype_ddp_optimization_mode)mmoder   r   r   _get_optimize_ddp_mode#  s   
r?   float64
complex128int64TORCHDYNAMO_CPP_GUARD_MANAGER%TORCHDYNAMO_INLINE_INBUILT_NN_MODULESc                  C   sR   d} | t jv rt jt j|  dS t r t jt t dS t jt 	 dS )NTORCH_COMPILE_DEBUG_DIRtorch_compile_debug)
osenvironpathjoinr   tempfile
gettempdirgetpassgetusergetcwd)DEBUG_DIR_VAR_NAMEr   r   r   default_debug_dir_rootw  s   
rQ   >   repro_afterrepro_levelconstant_functionsr    reorderable_logging_functions)striderequires_gradstorage_offsetlayoutdatac                 C   s    g | ]\}}t d |r|qS )z^is_.*)rematch).0name_r   r   r   
<listcomp>  s     r`    TORCH_FAKE_TENSOR_DISPATCH_CACHE+TORCH_FAKE_TENSOR_DISPATCH_CACHE_CROSSCHECK)*c                  K   s   d S )Nr   )changesr   r   r   _make_closure_patcher  s   re   )install_config_module)srM   inspectrG   r[   r4   rK   os.pathr   r   typingr   r   r   r   r   r	   r
   r   r   r   r   r9   __annotations__rH   getverboseverify_correctnessminimum_call_countdead_code_eliminationcache_size_limitaccumulated_cache_size_limit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   r8   suppress_errorsreplay_record_enabled rewrite_assert_with_torch_assertdisablecprofiler    *allowed_functions_module_string_ignorelistrR   intrS   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_matchr-   r/   r<   r?   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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_streamenable_cpp_guard_managerinline_inbuilt_nn_modulesrQ   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_metricsrU   +inject_BUILD_SET_unimplemented_TESTING_ONLY)_autograd_backward_strict_mode_banned_opsextend
getmembersTensorfake_tensor_cache_enabled$fake_tensor_cache_crosscheck_enabled:_experimental_support_context_fn_in_torch_utils_checkpointcompiled_autogradtorch.utils._config_typingre   torch.utils._config_modulerf   r5   r6   r   r   r   r   <module>   s   
(

