Ë
    �…jÄ  ã                   ó²   — d dl Z d dlZd dlZ G d„ d«      Z G d„ de«      Z G d„ de«      Z G d„ d	e«      Z G d
„ de«      Z G d„ de«      Z	 G d„ d«      Z
y)é    Nc                   ó   — e Zd Zd„ Zd„ Zy)Ú
BaseObjectc                 ó   — i | _         y ©N)Ú
customized©Úselfs    úa/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/onnxruntime/transformers/metrics.pyÚ__init__zBaseObject.__init__   s	   € Øˆ�ó    c                 ób  — | j                   j                  «       }|j                  dd «       |j                  | j                  «       |j                  «       D ])  \  }}t        |t        «      sŒ|j                  «       ||<   Œ+ |j                  «       D ��ci c]  \  }}|sŒ	||“Œ c}}S c c}}w )Nr   )	Ú__dict__ÚcopyÚpopÚupdater   ÚitemsÚ
isinstancer   Úto_dict)r	   Údefault_valuesÚkÚvs       r
   r   zBaseObject.to_dict   s•   € ØŸ™×+Ñ+Ó-ˆØ×Ñ˜<¨Ô.Ø×Ñ˜dŸo™oÔ.à"×(Ñ(Ó*ò 	0‰DˆAˆqÜ˜!œZÕ(Ø$%§I¡I£K�˜qÒ!ð	0ð "0×!5Ñ!5Ó!7×=™˜˜Aº1��1‘Ó=Ð=ùÓ=s   Â
B+Â"B+N)Ú__name__Ú
__module__Ú__qualname__r   r   © r   r
   r   r      s   „ òó	>r   r   c            
       óR   ‡ — e Zd Z	 	 	 	 ddedz  dedz  dedz  dedz  fˆ fd„Zˆ xZS )Ú	ModelInfoNÚ	full_nameÚis_huggingfaceÚis_text_generationÚ
short_namec                 óh   •— t         ‰| �  «        || _        || _        || _        || _        g | _        y r   )Úsuperr   r   r   r    r!   Úinput_shape)r	   r   r   r    r!   Ú	__class__s        €r
   r   zModelInfo.__init__   s6   ø€ ô 	‰ÑÔØ"ˆŒØ,ˆÔØ"4ˆÔØ$ˆŒØˆÕr   )NFFN)r   r   r   ÚstrÚboolr   Ú__classcell__©r%   s   @r
   r   r      sQ   ø„ ð !%Ø&+Ø*/Ø!%ñà˜‘:ðð ˜t™ðð ! 4™Kð	ð
 ˜$‘J÷ñ r   r   c                   óF   ‡ — e Zd Z	 	 	 ddedz  dedz  dedz  fˆ fd„Zˆ xZS )ÚBackendOptionsNÚenable_profilingÚexecution_providerÚuse_io_bindingc                 óL   •— t         ‰| �  «        || _        || _        || _        y r   )r#   r   r,   r-   r.   )r	   r,   r-   r.   r%   s       €r
   r   zBackendOptions.__init__.   s(   ø€ ô 	‰ÑÔØ 0ˆÔØ"4ˆÔØ,ˆÕr   )FNF)r   r   r   r'   r&   r   r(   r)   s   @r
   r+   r+   -   s@   ø„ ð ).Ø)-Ø&+ñ		-à ™+ð	-ð   $™Jð	-ð ˜t™÷		-ñ 	-r   r+   c                   ój   ‡ — e Zd Z	 	 	 	 	 	 d	dedz  dedz  dedz  dedz  dedz  dedz  fˆ fd„Zˆ xZS )
ÚConfigÚbackendNÚ
batch_sizeÚ
seq_lengthÚ	precisionÚwarmup_runsÚmeasured_runsc                 ó²   •— t         ‰| �  «        || _        || _        || _        || _        || _        || _        t        «       | _	        t        «       | _        y r   )r#   r   r2   r3   r4   r5   r6   r7   r   Ú
model_infor+   Úbackend_options)r	   r2   r3   r4   r5   r6   r7   r%   s          €r
   r   zConfig.__init__;   sP   ø€ ô 	‰ÑÔØˆŒØ$ˆŒØ$ˆŒØ"ˆŒØ&ˆÔØ*ˆÔÜ#›+ˆŒÜ-Ó/ˆÕr   )Úonnxruntimeé   r   Úfp32r<   é
   )r   r   r   r&   Úintr   r(   r)   s   @r
   r1   r1   :   ss   ø„ ð ,Ø!"Ø!"Ø &Ø"#Ø$&ñ0à�t‘ð0ð ˜$‘Jð0ð ˜$‘Jð	0ð
 ˜‘:ð0ð ˜4‘Zð0ð ˜T‘z÷0ñ 0r   r1   c                   ó^   ‡ — e Zd Z	 	 	 	 	 ddedz  dedz  dedz  dedz  dedz  f
ˆ fd„Zˆ xZS )	ÚMetadataNÚdeviceÚpackage_nameÚpackage_versionÚplatformÚpython_versionc                 óh   •— t         ‰| �  «        || _        || _        || _        || _        || _        y r   )r#   r   rB   rC   rD   rE   rF   )r	   rB   rC   rD   rE   rF   r%   s         €r
   r   zMetadata.__init__P   s6   ø€ ô 	‰ÑÔØˆŒØ(ˆÔØ.ˆÔØ ˆŒØ,ˆÕr   )NNNNN)r   r   r   r&   r   r(   r)   s   @r
   rA   rA   O   sb   ø„ ð "Ø#'Ø&*Ø#Ø%)ñ-à�d‘
ð-ð ˜D‘jð-ð ˜t™ð	-ð
 ˜‘*ð-ð ˜d™
÷-ñ -r   rA   c                   óF   ‡ — e Zd Z	 	 	 ddedz  dedz  dedz  fˆ fd„Zˆ xZS )ÚMetricsÚlatency_ms_meanNÚthroughput_qpsÚmax_memory_usage_GBc                 óL   •— t         ‰| �  «        || _        || _        || _        y r   )r#   r   rJ   rK   rL   )r	   rJ   rK   rL   r%   s       €r
   r   zMetrics.__init__a   s(   ø€ ô 	‰ÑÔØ.ˆÔØ,ˆÔØ#6ˆÕ r   )ç        rN   rN   )r   r   r   Úfloatr   r(   r)   s   @r
   rI   rI   `   s@   ø„ ð ),Ø'*Ø,/ñ		7à ™ð	7ð  ™ð	7ð # T™\÷		7ñ 	7r   rI   c                   ó²   — e Zd Z	 	 	 	 ddedededededededz  d	edz  d
edz  dedz  fd„Zdefd„Zdefd„Ze	dede
ddfd„«       Ze	dede
ddfd„«       Zy)ÚBenchmarkRecordNÚ
model_namer5   r2   rB   rC   rD   r3   r6   r7   Útrigger_datec                 ó  — t        «       | _        t        «       | _        t	        «       | _        |
xs- t        j                  j                  «       j                  d«      | _	        || j                  j                  _        || j                  _        || j                  _        || j                  _        || j                  _        |	| j                  _        || j
                  _        || j
                  _        || j
                  _        y )Nz%Y-%m-%d %H:%M:%S)r1   ÚconfigrI   ÚmetricsrA   ÚmetadataÚdatetimeÚnowÚstrftimerS   r9   r   r5   r2   r3   r6   r7   rB   rC   rD   )r	   rR   r5   r2   rB   rC   rD   r3   r6   r7   rS   s              r
   r   zBenchmarkRecord.__init__n   s¹   € ô “hˆŒÜ“yˆŒÜ ›
ˆŒØ(Òa¬H×,=Ñ,=×,AÑ,AÓ,C×,LÑ,LÐM`Ó,aˆÔà+5ˆ�‰×ÑÔ(Ø )ˆ�‰ÔØ%ˆ�‰ÔØ!+ˆ�‰ÔØ"-ˆ�‰ÔØ$1ˆ�‰Ô!Ø%ˆ�‰ÔØ%1ˆ�‰Ô"Ø(7ˆ�‰Õ%r   Úreturnc                 ó´   — | j                   j                  «       | j                  j                  «       | j                  j                  «       | j                  dœS )N)rU   rW   rV   rS   )rU   r   rW   rV   rS   r   s    r
   r   zBenchmarkRecord.to_dictŠ   sE   € à—k‘k×)Ñ)Ó+ØŸ™×-Ñ-Ó/Ø—|‘|×+Ñ+Ó-Ø ×-Ñ-ñ	
ð 	
r   c                 óT   — t        j                  | j                  «       t        ¬«      S )N)Údefault)ÚjsonÚdumpsr   r&   r   s    r
   Úto_jsonzBenchmarkRecord.to_json’   s   € Ü�z‰z˜$Ÿ,™,›.´#Ô6Ð6r   Ú	file_nameÚrecordsc                 óº   — |�t        |«      dk(  ry |D �cg c]  }|j                  «       ‘Œ }}t        j                  |«      }|j	                  |d¬«       y c c}w )Nr   F)Úindex)Úlenr   ÚpdÚjson_normalizeÚto_csv)Úclsrb   rc   ÚrecordÚrdsÚdfs         r
   Úsave_as_csvzBenchmarkRecord.save_as_csv•   sT   € àˆ?œc '›l¨aÒ/ØØ.5Ö6 Fˆv�~‰~ÕÐ6ˆÐ6Ü×Ñ˜sÓ#ˆØ
�	‰	�) 5ˆ	Õ)ùò 7s   –Ac                 óæ   — |�t        |«      dk(  ry |D �cg c]  }|j                  «       ‘Œ }}t        |d«      5 }t        j                  ||dt
        ¬«       d d d «       y c c}w # 1 sw Y   y xY w)Nr   Úwé   )Úindentr^   )rf   r   Úopenr_   Údumpr&   )rj   rb   rc   rk   rl   Úfs         r
   Úsave_as_jsonzBenchmarkRecord.save_as_json�   sl   € àˆ?œc '›l¨aÒ/ØØ.5Ö6 Fˆv�~‰~ÕÐ6ˆÐ6Ü�)˜SÓ!ð 	5 QÜ�I‰I�c˜1 Q´Õ4÷	5ð 	5ùò 7÷	5ð 	5ús   –A"»A'Á'A0)r<   r<   r>   N)r   r   r   r&   r?   r   Údictr   ra   ÚclassmethodÚlistrn   rv   r   r   r
   rQ   rQ   m   sñ   „ ð "#Ø"#Ø$&Ø#'ñ8àð8ð ð8ð ð	8ð
 ð8ð ð8ð ð8ð ˜$‘Jð8ð ˜4‘Zð8ð ˜T‘zð8ð ˜D‘jó8ð8
˜ó 
ð7˜ó 7ð ð* Cð *°$ð *¸4ò *ó ð*ð ð5 Sð 5°4ð 5¸Dò 5ó ñ5r   rQ   )rX   r_   Úpandasrg   r   r   r+   r1   rA   rI   rQ   r   r   r
   ú<module>r{      s`   ðó Û ã ÷>ñ >ô �
ô ô 
-�Zô 
-ô0ˆZô 0ô*-ˆzô -ô"
7ˆjô 
7÷65ò 65r   