P/j5Q |UddlZddlZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZmZmZddlZdZdZe eed<ej.j1dddk(Zd Zd Zd Zd Zd Zd Zd Z d Z!d Z"ej.j1dddk(Z#d Z$d Z%d Z&d Z'd Z(d Z)d Z*e+Z,eee ed<e-ej.j1dd Z.ej.j1dddk(Z/d Z0ej.j1dd Z1ej.j1dd Z2iZ3e e e fed<hdZ4ej.j1ddZ5e6ej.j1ddZ7ej.j1ddk(Z8dZ9ej.j1ddk(Z:d Z;ej.j1ddk(Zd Z?d Z@d ZAd ZBee-efed<ej.j1dddk(ZCe-ed <gd!ZDd"ZEd ZFd ZGd ZHd ZId ZJd ZKd ZLd ZMd ZNd ZOerd nd ZPd ZQd ZReeeeeSZTd ZUd#ZVd$ZWd%ZXd ZYej.j1d&ddk(ZZe Z[d Z\d'Z]e]Z^hd(Z_d Z`d Zad Zbd Zcd Zdd Zee+Zfee e gdfed)<d Zggd*ZhehjejejDcgc]\}}ejd+|s|c}}ej.j1d,ddk(Zmej.j1d-ddk(Znd Zoej.j1d.ddk(Zperdd/lqd0Zrdd1lsmtZtetejevycc}}w)2N)abspathdirname)AnyCallableDictOptionalSetType TYPE_CHECKINGUnionc8ttjd S)N git_version)hasattrtorchversionW/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/_dynamo/config.py is_fbcodersu}}m4 44r log_file_nameTORCHDYNAMO_VERBOSE01FT#TORCH_COMPILE_USE_LAZY_GRAPH_MODULEtraceable_tensor_subclassesTORCHDYNAMO_SUPPRESS_ERRORSTORCH_COMPILE_REPLAY_RECORDTORCH_COMPILE_DISABLETORCH_COMPILE_CPROFILE!skipfiles_inline_module_allowlist> torch._refs torch._prims torch._decomp torch.testingtorch.distributionsTORCHDYNAMO_REPRO_AFTERTORCHDYNAMO_REPRO_LEVELTORCHDYNAMO_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_ASSERTSdo_not_emit_runtime_asserts) ddp_optimizerpython_reducer'python_reducer_without_compiled_forwardno_optimizationc@tjt}t|jt r|jrd}nEd}nBt|jt r |j}ntdtt||jvs Jd||S)Nr3r6z!Invalid type, type(optimize_ddp)=zInvalid mode mode=) sysmodules__name__ isinstancer0boolstr ValueErrortype_ddp_optimization_mode)mmodes r_get_optimize_ddp_moderC&s HA!..$' >>"D$D ANNC (~~=$|*<)>?@@ 1++ +D/BTG-DD + Krfloat64 complex128int64TORCHDYNAMO_CPP_GUARD_MANAGERcd}|tjvr1tjjtj|dSt rEtjjt j tjdStjjtjdS)NTORCH_COMPILE_DEBUG_DIRtorch_compile_debug) osenvironpathjoinrtempfile gettempdirgetpassgetusergetcwd)DEBUG_DIR_VAR_NAMEs rdefault_debug_dir_rootrUs2RZZ'ww||BJJ'9:??r> repro_after repro_levelconstant_functionsr#reorderable_logging_functions)stride requires_gradstorage_offsetlayoutdataz^is_.* TORCH_FAKE_TENSOR_DISPATCH_CACHE+TORCH_FAKE_TENSOR_DISPATCH_CACHE_CROSSCHECKTORCH_COMPILER_COLLECTIVES)*c y)Nr)changess r_make_closure_patcherres r)install_config_module)wrQinspectrKrer8rOos.pathrrtypingrrrrr r r r rrrr=__annotations__rLgetverboseverify_correctnessminimum_call_countdead_code_eliminationcache_size_limitaccumulated_cache_size_limit&skip_code_recursive_on_cache_limit_hitspecialize_intspecialize_floatdynamic_shapesuse_lazy_graph_moduleassume_static_by_defaultautomatic_dynamic_shapesforce_parameter_static_shapes&force_nn_module_property_static_shapesallow_ignore_mark_dynamicguard_nn_modules guard_nn_modules_using_dict_tagssetrr<suppress_errorsreplay_record_enabled rewrite_assert_with_torch_assertdisablecprofiler#*allowed_functions_module_string_ignorelistrVintrWrepro_forward_onlyrepro_tolerancerepro_ignore_non_fpsame_two_models_use_fp64capture_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_kjtenforce_cond_guards_matchr0r2r@rC skip_torchrecoptimize_ddp_lazy_compileskip_fsdp_guardsskip_fsdp_hooksskip_nnmodule_hook_guardsraise_on_ctx_manager_usageraise_on_unsafe_aot_autograderror_on_nested_jit_traceerror_on_nested_fx_trace allow_rnncapture_sparse_computeerror_on_recompilereport_guard_failures__file__base_dir trace_numpynumpy_default_floatnumpy_default_complexnumpy_default_intuse_numpy_random_streamenable_cpp_guard_managerinline_inbuilt_nn_modules%record_compile_time_instruction_countrUdebug_dir_root_save_config_ignore%cudagraph_backend_keep_input_mutation(cudagraph_backend_support_input_mutationonly_allow_pt2_compliant_opscapture_autograd_functioncapture_func_transformslog_compilation_metricsrY+inject_BUILD_SET_unimplemented_TESTING_ONLY)_autograd_backward_strict_mode_banned_opsextend getmembersTensormatchfake_tensor_cache_enabled$fake_tensor_cache_crosscheck_enabledcompiled_autogradenable_compiler_collectivestorch.utils._config_typingretorch.utils._config_modulerfr9r:)name_s00rrs $QQQ 5 $ x}# **... 4 ; #*.& JJNN8#>#E    !% *.&"$( &/2eSc^3 rzz~~&CUKL  'DcJcQ$(  **..0% 8 ::>>2E :57!4S>6 .*jjnn6= "**..!:A>? ZZ^^$DEL jjnn%FG3N  (LMQTTJJNNA3G3N! /4+ +0'7<3!*"& eD#I%JJNN=sCsJT " " !" %!   #,+4  7778#456 7  $ ::>>*I3OSVV!* O).% @()).%,1( %  =@Es8SE4K#89A/4+-)*00+++ELL9WgdAXRXXiQU=VTW JJNN5s;sB JJNN@#F#M% !jjnn-I3OSVV, =ckk(+,MXs N8 'N8