Ph'BddlZddlmZddlmZmZmZmZddlZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZdd lmZddlmcmZddlZed d Zed dZ ejBe"Z#de$de%fdZ&dee$ee$e$fffdZ'GddZ(y)N) defaultdict)TupleDictOptionalList)export)_ExportPassBase) NodeMetadata) ProxyValue) FakeTensor)TargetArgument)Library)tree_unflattenatenFRAGMENTIMPLversioned_upgrader_namereturnctjd|std|dt|j dddzS)zdiv_Scalar_0_3 is the name of the upgrader, meaning it applies to div.Scalar of version 0 to 3 and is upgrading to version 4.z^.*_[0-9]+_[0-9]+$zUpgrader name z is invalid_)rematch RuntimeErrorintsplit)rs fC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_export/serde/upgrade.pyget_target_versionr sK 88(*A B^,C+DKPQQ &,,S1"5 6 ::c~tjj}tjj}t t }|j D]^\}}|std||d}|j}|j}|j|d}|std|d|||f||<`|S)zRGetting upgraders entry map and operator version map and merge them into one dict.z-Op version map has an empty entry for opname rNzCan't find upgrader for op z and upgrader name ) torch_C_get_upgraders_entry_map_get_operator_version_maprtupleitemsr old_schema upgrader_nameget) upgradersop_version_mapoutputopname entry_listentryr)r* upgrader_strs r get_upgradersr3 s113IXX779N)4U);F,224 !NvhWX X1 %% ++  }}]D9 ! torch.Tensor: # upgrader in literal string if (self.is_floating_point() or isinstance(other, float)): return self.true_divide_(other) return self.divide_(other, rounding_mode='trunc') ''', ), }, Note that we require the upgrader function to be runnable in Python (which is a stricter requirement than the original TorchScript upgrader). cXeZdZdedeffd Zdeedfdeeefde de ffd Z xZ S) "GraphModuleOpUpgrader.UpgraderPass old_target new_targetc>t|||_||_yN)super__init__r8r9)selfr8r9 __class__s rr=z+GraphModuleOpUpgrader.UpgraderPass.__init__Ns G  (DO(DOr!args.kwargsmetarc~||jk(rt| |j|||St| ||||Sr;)r8r< call_operatorr9)r>opr@rArBr?s rrDz0GraphModuleOpUpgrader.UpgraderPass.call_operatorSsAT__$w,T__dFDQQ7(T64@ @r!) __name__ __module__ __qualname__r r=rrrstrr r rD __classcell__)r?s@r UpgraderPassr7Ms` )v )6 ) AHcM* AS(]+ A # A  A Ar!rKNcompiler_opset_versionmodel_opset_version op_upgradersc|s tn||_|r|ni|_|r|ni|_tj |j |j|_yr;)r3rNrLrMr5_populate_passes_parse_upgradersupgrader_passes)r>rLrMrNs rr=zGraphModuleOpUpgrader.__init__^sY P\ am@V&<\^#:M#6SU I^IoIo  ! !$"3"3 4J6r!rcd}|r||jvs||jvrgS|j|}|j|}|jDcic]\}}t||}}}g}t |dz|dzD]4} | |vr|j || t jdd| i6|Scc}}w)zReorder op_upgraders by version number, return an ordered list of tuples, containing old op schema as well as the upgrader function string literal.rrz0Missing an upgrader to upgrade to version {ver}.ver)extra)rMrLr(r rangeappendlogwarning) r>rN op_namespace model_vercurr_vernamevversioned_upgraderstarget_upgradersrTs rrQz&GraphModuleOpUpgrader._parse_upgradersjs |43K3KK|cgc~c~O~I,,\: ..|<__. Register upgraders as CompositeImplicitAutograd kernels. For example: lib = Library("aten", "FRAGMENT") lib.define(old_schema) impl_lib = Library("aten", "IMPL") impl_lib.impl("div__Scalar_0_3", div__Scalar_0_3, "CompositeImplicitAutograd") @:var upgraders: a list of tuples. The first element of the tuple is the old schema and the second is the upgrader function literal text. @:return upgrader passes, order matters r]schemaimpl_strctj| t|t j |t|dy#t$r}t d||d}~wwxYw)zARegisters an old version operator using impl_name as old op name.zInvalid upgrader string: NCompositeImplicitAutograd)libdefineexec Exceptionrimpl_libimpllocals)r]rbrces rregister_old_opz?GraphModuleOpUpgrader._populate_passes..register_old_ops^ JJv  RX MM$0K L R"%>xj#IJPQQ Rs A A"AA"(r rz::)r]rbrcz3with the same name and overload name multiple timesz Registering z multiple timesN.default)r8r9)rIrreplacerprintgetattrr#opsrrrr'rWr5rK) r,rRrnrbr2r*op_namerm old_op_target overload_name new_op_targets rrPz&GraphModuleOpUpgrader._populate_passessx" M# Ms Mc M'0 "V\(..s3A6<A=Ogy%9UZ[b[h[hil[mnpop[qUr "G]#GEIINNG$DmTM  " "%22mXe2f h#'0(  .HCPQFRLGH&A-I .s D;; E2$E--E2exported_programc ||js|S|jjDcgc].}|jdk(s|jj dd0}}|Dcgc]L}t |tr8tjt|j|jn|N}}|jjJt||jj\}}|ik(sJ|jD].}|j!|}t#|j%||}0|Scc}wcc}w)aRun each upgrader pass and then retrace to decompose it. Each upgrader pass replaces the old version of operators with a custom operator. The custom operator contains a CompositeImplicitAutograd kernel (the upgrading function itself). After retrace, this custom operator will be decomposed into the ops used in the upgrader. After all passes are applied, the exported program will be upgraded to the target version. placeholdervalN)dtype)rRgraphnodesrErBr+ isinstancer r#onesr'sizer call_specin_specr _transformrmodule) r>r|nr@argargs_real_tensorsrA_passupgraded_programs rupgradezGraphModuleOpUpgrader.upgrades/ ### #1A1G1G1M1Mg1MAQRQUQUYfQf 5$'1Mg$(*$(SPZZ]_iOjUZZchhj(9Kpss$( *))11===%&79I9S9S9[9[\ f||))E/::5A %&6&=&=&?vN *  h*sD4D4 AD9)NNNr;)rFrGrH__doc__r rKrrrIrrr=rrQ staticmethodrPepExportedProgramrr!rr5r52s4AA&@D<@AE 6$,T#s(^$< 6"*$sCx.!9 6#4U38_(<#=> 6 Xd3c3h;O6P-Q ]abghkmphpbq]r 2/DsCx$9/d<>P//b (:(: r?Q?Q r!r5))logging collectionsrtypingrrrrr# torch._exportrtorch._export.pass_baser &torch._export.pass_infra.node_metadatar $torch._export.pass_infra.proxy_valuer torch._subclassesr torch.fx.noder r torch.libraryrtorch.utils._pytreertorch._export.exported_program_exportr|rrrfrj getLoggerrFrXrIrr r3r5rr!rrs#.. 3?;(*!.++  fj! 66 "g!;;;tCsCx01$W W r!