U <c*@sNdZddlmZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZGdddejZeZGdd d ejZejd d Gd d d ZejGdddZejGdddZedddZejGdddZejGdddZejd d GdddZGdddZejGdddZejGdddZ dS) zLThis file defines an additional layer of abstraction on top of the SARIF OM.) annotationsN) Any FrozenSetListOptionalSequenceSetTupleTypeTypeVar) formattersarifc@s eZdZdZdZdZdZdZdS)LevelzThe level of a diagnostic. This class is used to represent the level of a diagnostic. The levels are defined by the SARIF specification, and are not modifiable. For alternative categories, please use infra.Tag instead. noneZnotewarningerrorN)__name__ __module__ __qualname____doc__NONEZNOTEWARNINGERRORrrQ/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/_internal/diagnostics/infra/_infra.pyr s rc@seZdZdZdS)TagzKThe tag of a diagnostic. This class can be inherited to define custom tags.NrrrrrrrrrsrT)frozenc@sjeZdZUded<ded<ded<dZded<dZded<dZded <edd d d Zd d ddZ dS)Rulestridnamemessage_default_templateN Optional[str]short_descriptionfull_descriptionhelp_urireturncKspd|kr|ddnd}d|kr,|ddnd}d|kr@|dnd}||d|d|d d d|||d }|S) z3Returns a rule from the SARIF reporting descriptor.r$textNr%markdownr&r r!Zmessage_stringsdefault)r r!r"r$r%r&r)clskwargsr$r%r&rulerrr from_sarif,s$zRule.from_sarifzsarif.ReportingDescriptorcCsT|jdk rtj|jdnd}|jdk r6tjd|jdnd}tj|j|j|||jdS)z2Returns a SARIF reporting descriptor of this Rule.Nr))r)r*)r r!r$r%r&)r$r ZMultiformatMessageStringr%ZReportingDescriptorr r!r&)selfr$r%rrrr Esz Rule.sarif) rrr__annotations__r$r%r& classmethodr/r rrrrr#s    rc@sPeZdZUded<ded<dZded<dZded<dZded<d d d d ZdS) LocationrurimessageN Optional[int]line start_column end_columnzsarif.Locationr'c CsBtjtjtj|jdtj|j|j|j|jddtj |j ddS)z2Returns the SARIF representation of this location.)r6)Z start_liner:Zend_liner;)Zartifact_locationZregionr0)Zphysical_locationr7) r r5ZPhysicalLocationZArtifactLocationr6ZRegionr9r:r;Messager7r2rrrr bs  zLocation.sarif)rrrr3r9r:r;r rrrrr5Zs    r5c@sJeZdZUejedZded<ddddZdd d d d d d d d dZ dS)Stackdefault_factoryList[Location]frame_locationsz sarif.Stackr'cCstjdd|jDdS)z/Returns the SARIF representation of this stack.cSsg|]}tj|dqS))location)r Z StackFrame).0locrrr yszStack.sarif..)frames)r r>rBr=rrrr vs z Stack.sarifNrr8None)r6r7r9r:r;r(c Cs|jt|||||ddS)zAdds a frame to the stack.)r6r7r9r:r;N)rBappendr5)r2r6r7r9r:r;rrr add_frame~s zStack.add_frame)NNN) rrr dataclassesfieldlistrBr3r rJrrrrr>rs  r> _Diagnostic Diagnostic)boundc@seZdZUded<ded<ded<ejedZded <ejedZd ed <d Z d ed<ejedZ ded<ddddZ ddddddZ ddddddZ ddddd d!Zd S)"rOrr.rlevelOptional[Tuple[Any, ...]] message_argsr?rA locationsz List[Stack]stacksNr#additional_messagez List[Tag]tagsz sarif.Resultr'cCs|jdkrt|_|jjj|j}|jdk r<|d|j}tjtj|d|j j |jj d}dd|j D|_ dd|j D|_ tjdd|jDd |_|S) z;Returns the SARIF Result representation of this diagnostic.N r0)r7rQZrule_idcSsg|] }|qSrr )rDrCrrrrFsz$Diagnostic.sarif..cSsg|] }|qSrrY)rDstackrrrrFscSsg|] }|jqSr)value)rDtagrrrrFs)rW)rStupler.r"formatrVr ZResultr<rQr[r rTrUZ PropertyBagrWZ properties)r2r7Z sarif_resultrrrr s    zDiagnostic.sarifrNr5)r2rCr(cCs|j||S)z"Adds a location to the diagnostic.)rTrI)r2rCrrr with_locations zDiagnostic.with_locationr>)r2rZr(cCs|j||S)zAdds a stack to the diagnostic.)rUrI)r2rZrrr with_stacks zDiagnostic.with_stackr)r2r7r(cCs(|jdkr||_n|jd||_|S)z-Adds an additional message to the diagnostic.NrX)rV)r2r7rrrwith_additional_messages z"Diagnostic.with_additional_message)rrrr3rKrLrMrTrUrVrWr r_r`rarrrrrOs  c@sVeZdZUejddZded<ddddZd d d d d Ze ddddddZ dS)RuleCollectionFinitzFrozenSet[Tuple[str, str]]_rule_id_name_setrHr'cCstddt|D|_dS)NcSs(h|] }t|jtr|jj|jjfqSr) isinstancer+rr r!)rDrLrrr s z/RuleCollection.__post_init__..) frozensetrKfieldsrer=rrr __post_init__s zRuleCollection.__post_init__rbool)r.r(cCs|j|jf|jkS)z(Checks if the rule is in the collection.)r r!re)r2r.rrr __contains__szRuleCollection.__contains__rzSequence[Rule])new_collection_class_namerulesr(cCstj|dd|D|fdS)zLCreates a custom class inherited from RuleCollection with the list of rules.cSs*g|]"}t|jt|tj|dfqS)r+)r Zkebab_case_to_snake_caser!typerKrLrDr.rrrrFs   z>RuleCollection.custom_collection_from_list..)bases)rKZmake_dataclass)r,rmrnrrrcustom_collection_from_listsz*RuleCollection.custom_collection_from_listN) rrrrKrLrer3rjrlr4rsrrrrrbs  rbc@seZdZUded<ded<ded<ejedZded<ejd ed Z d ed <d dddZ ddddZ dddddddZ dS)DiagnosticToolrr!versionrbrnrozType[Diagnostic]diagnostic_typeFrdr@z Set[Rule]_triggered_rulesrHr'cCs t|jtstd|jdS)NzAExpected diagnostic_type to be a subclass of Diagnostic, but got ) issubclassrvrO TypeErrorr=rrrrjs  zDiagnosticTool.__post_init__z sarif.ToolcCs(tjtj|j|jdd|jDddS)z&Returns the SARIF Tool representation.cSsg|] }|qSrrYrqrrrrFsz(DiagnosticTool.sarif..)r!rurn)Zdriver)r ZToolZ ToolComponentr!rurxr=rrrr szDiagnosticTool.sarifrrrRrOr.rQrSr(c KsZ||jkr.)rresults)r ZRunrrr=rrrr ;szDiagnosticContext.sarifrrrRrOr{cKs$|jj|||f|}|j||S)r|)rrrrI)r2r.rQrSr-rrrrdiagnoseBs zDiagnosticContext.diagnose)N)rrrr3rrKrLrMrrrrr rrrrrr.s   r)!r __future__rrKenumtypingrrrrrrr r r Z&torch.onnx._internal.diagnostics.infrar r EnumrlevelsrZ dataclassrr5r>rNrOrbrtrrrrrrrs4 , 6 /# 6