o iV@sZddlZddlZddlZddlZddlZddlZddlZdefddZddZ ddZ dS) NreturncCstjr tjStS)N)torchcuda is_availableget_device_nameplatformmachiner r I/home/ubuntu/.local/lib/python3.10/site-packages/torchao/_models/utils.py get_arch_names  r csfddt|D}dd|d|d|d|d|d d d |d d dgd|d|dg|ddd}ttj|ddd}tt||dWddS1sZwYdS)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 ci|] \}}||qSr r .0ivheadersr r z+write_json_result_ossci..TorchAO benchmark inferencedtypedevicearchmin_sqnrcompilerrrrnamemoder extra_informodeltorchaortypeoriginsmetricactualtargetrbenchmark_values target_value) benchmarkr!r&r.jsonafileN) enumerateopenospathsplitextprintjsondumps)output_json_pathrrowmapping_headersrecordfr rr write_json_result_osscis, "r>c s$fddt|D}tj}tt|d}tj |tj j }t | }|d}t t dddd||d|d d |d |d |d |d|ddd|dddgd|d|dg|ddd } ttj|ddd} tt| | dWddS1swYdS) 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) cr r r r rr r rFrz+write_json_result_local..zutf-8v3zdevvm local benchmarkz pytorch/aomainrrrrrrrrrrr!r"r#r&r'r(r)) timestampschema_versionrrepo head_branchhead_sha workflow_id run_attemptjob_idr,r!r&rr-r.r/N)r1datetimedatetodayhashlibsha256strencode hexdigestcombinetimeminintrBr2r3r4r5r6r7r8) r9rr:r;rLsha_hash first_secondrGrIr<r=r rr write_json_result_local>sH    ""rX) rJrMr7r3rrSrrOr r>rXr r r r s &