PiVVddlZddlZddlZddlZddlZddlZddlZdefdZdZ dZ dS)NreturnctjrtjSt jS)N)torchcuda is_availableget_device_nameplatformmachinei/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchao/_models/utils.py get_arch_namers= z  "z))+++!!!r cfdt|D}dd|d|d|d|d|dd d |d d d gd|d|dg|ddd}ttj|ddd5}t t j||ddddS#1swxYwYdS)aM Write the result into JSON format, so that it can be uploaded to the benchmark database to be displayed on OSS dashboard. The JSON format is defined at https://github.com/pytorch/pytorch/wiki/How-to-integrate-with-PyTorch-OSS-benchmark-database OSS CI version, that will leave many fields to be filled in by CI c(i|]\}}||Sr r .0ivheaderss r z+write_json_result_ossci.. #@@@Awqz1@@@r TorchAO benchmark inferencedtypedevicearchmin_sqnrcompilerrrrnamemoder extra_infor!modeltorchaor!typeoriginsmetricactualtargetr!benchmark_values target_value) benchmarkr$r)r.jsonafileN) enumerateopenospathsplitextprintjsondumps)output_json_pathrrowmapping_headersrecordfs ` r write_json_result_osscirAsgA@@@3@@@O($W-)(3'/+J7*95    $F+!{  $H-!0!: ;+H5  %F2 !!"233A6===s C C*q dj  q))))******************s#$CCCcfdt|D}tj}t jt |d}tj |tj j }t| }|dz}ttj dddd||d|dd |d |d |d |d |ddd|dddgd|d|dg|ddd } ttj|ddd5} t%t'j| | ddddS#1swxYwYdS)a] Write the result into JSON format, so that it can be uploaded to the benchmark database to be displayed on OSS dashboard. The JSON format is defined at https://github.com/pytorch/pytorch/wiki/How-to-integrate-with-PyTorch-OSS-benchmark-database Local version (filling in dummy values for fields that should be populated by CI) c(i|]\}}||Sr r rs r rz+write_json_result_local..Frr zutf-8v3zdevvm local benchmarkz pytorch/aomainrrrrrrrrr r!r$r%r&r)r*r+r,) timestampschema_versionr!repo head_branchhead_sha workflow_id run_attemptjob_idr/r$r)rr0r1r2N)r4datetimedatetodayhashlibsha256strencode hexdigestcombinetimeminintrGr5r6r7r8r9r:r;) r<rr=r>rQsha_hash first_secondrLrNr?r@s ` r write_json_result_localr]>sA@@@3@@@O M   ! !E~c%jj//8899CCEEH$,,UHM4EFFLl,,..//K 1_F%%'"'$W-)(3'/+J7*95    $F+!{  $H-!0!: ;+H5  7  FD !!"233A6===s C C*q dj  q))))******************s$F44F8;F8) rOrRr:r6r rXrrTrrAr]r r r r^s   "s""""#*#*#*L1*1*1*1*1*r