DŽgddlZddlmZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZdZej(rdZd Zyej,ej.ej.eddZd Zy) N) backcompat collect_envdata deterministichooks)(generate_methods_for_privateuse1_backendrename_privateuse1_backend)get_cpp_backtrace)ThroughputBenchmarkcHt|ts td||_y)z[ Set the module attribute on a python object for a given object for nicer printing z#The mod argument should be a stringN) isinstancestr TypeError __module__)objmods \/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/utils/__init__.py set_modulers! c3 =>>CNsharecmakec tjr tdtjr tdtt j j }tt j j }||k7r tdfd}dd fd }|d|d|d |d |D]}t|rt|r ||$t|r$t|t|t|Tt|sat|t|t|tjjy ) z This function swaps the content of the two Tensor objects. At a high level, this will make t1 have the content of t2 while preserving its identity. This will not work if t1 and t2 have different slots. z8Cannot swap t1 because it has weakref associated with itz8Cannot swap t2 because it has weakref associated with itz2Cannot swap t1 and t2 if they have different slotscft|}t|t|t||y)N)getattrsetattr)nametmpt1t2s r swap_attrzswap_tensors..swap_attr;s.b$D72t,.D#rctd)NaTrying to execute AccumulateGrad node that was poisoned by swap_tensors this can happen when you try to run backward on a tensor that was swapped. For a module m with `torch.__future__.set_swap_module_params_on_conversion(True)` you should not change the device or dtype of the module (e.g. `m.cpu()` or `m.half()`) between running forward and backward. To resolve this, please only change the device/dtype before running forward (or after both forward and backward).) RuntimeError) grad_outputss rerror_pre_hookz$swap_tensors..error_pre_hook@s X  rrc@|j}d|d|d}|dkDr|dk(ro|jrctjjj |j }|jdk(r|jyt|t|y)NzExpected use_count of z2 to be 1 or 2 with an AccumulateGrad node but got zH make sure you are not holding references to the tensor in other places.) _use_countis_leaftorchautogradgraphget_gradient_edgenoderegister_prehookr")tr use_count error_straccum_grad_noder$s rcheck_use_countz%swap_tensors..check_use_countJsLLN $TF*\]f\ghV W  q=A~!))"'.."6"6"H"H"K"P"P<<>Q&#44^D&y11"9-- rr __class____dict__N)r)weakref getweakrefsr"setcopyreg _slotnamesr5hasattrrrdelattrr*_C_swap_tensor_impl)rrt1_slotst2_slotsr r4slotr$s`` @r swap_tensorsrC)sG2UVV2UVV7%%bll34H7%%bll34H8OPP  ."BBkj 2t T!2 dO R  Bwr40 2 B  R  Bwr40 2 B  HHr2&r)r:os.pathpath_ospr7r* torch.utilsrrrrr torch.utils.backend_registrationrr torch.utils.cpp_backtracer torch.utils.throughput_benchmarkr r_running_with_deploycmake_prefix_pathjoindirname__file__rCrrrQs 8@5H' "  \T\\(+,gw H'r