IЦi3 @SSKrSSKJr SSKJrJrJr SSKrSSKJr \R"\ 5r Sr Sr "SS\5rS rS \\\R"\\R"S 44S \\\R"\\R"S 444S jr\"SS55rg)N) dataclass)ListTupleUnion)fxc\^/mU4Sjn[RRUU5nUT4$)zT Flatten the args into a list form and detach the tensors from computational graph. c>[U[R5(a<UR5R UR 5nTR U5 U$TR U5 U$N) isinstancetorchTensordetachrequires_grad_ requires_gradappend)avalflat_detached_argss b/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/distributed/pipelining/_utils.pyextract_tensor_args0flatten_args_detach..extract_tensor_argssT a & &((*++AOOTRU5 U$r )r)r flat_argss rr)flatten_args..extract_tensor_args,srr)rrr!s @r flatten_argsr#&s1I GG  rc\rSrSrSrSrg)PipeliningShapeError9z5Shape mismatch between configured and runtime values.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r'rrr%r%9s?rr%cURUR:Xd'[USURSUR35eURUR:Xd'[USURSUR35eUR5UR5:Xd/[USUR5SUR535eg)Nz has a shape mismatch: expected z actual z has a dtype mismatch: expected z! has a stride mismatch: expected )shaper%dtypestride)descexpectedgivens rvalidate_tensor_metadatar5=s >>U[[ ("f4X^^4DHU[[M Z   >>U[[ ("f4X^^4DHU[[M Z   ??   ."f5hoo6G5HQVQ]Q]Q_P` a   /rexpected_tensors.actual_tensorsc [U5[U5:wa&[US[U5S[U5S35e[[U55Hn[USU3XX#5 M g)Nz: Number of values (z") does not match expected number ()z: value )lenr%ranger5)r2r6r7is rvalidate_tensors_metadatar=Ls  N 33"f(^)<(==_`cdt`u_vvw x  3'( ) fHQC "2"5~7H *rcL\rSrSr%Sr\R \S'\\S'\ \S'Sr g)PipeInfo[z6 Captures information for a pipeline (`Pipe` object). graph num_stageshas_loss_and_backwardr'N) r(r)r*r+r,rGraph__annotations__intboolr-r'rrr?r?[s 88OOrr?)logging dataclassesrtypingrrrr r getLoggerr(loggerrr# RuntimeErrorr%r5r r=r?r'rrrNs!%%    8 $(0&@<@    D.ellC6G0HHI  $u||,eELL#4E.FFG       r