hVVSSKrSSKrSSKrSSKrSSKrSSKrSSKrS\4SjrSr Sr g)Nreturnc[RR5(a[RR5$[R "5$)N)torchcuda is_availableget_device_nameplatformmachineO/home/james-whalen/.local/lib/python3.13/site-packages/torchao/_models/utils.py get_arch_namers9 zz  zz))++!!r c[U5VVs0sH up4XU_M nnnSSUSUSUSUSUSS.S .US S S /S .USUS/USS.S.n[[RR U5SS3S5n[ [ R"U5US9 SSS5 gs snnf!,(df  g=f)a9 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 TorchAO benchmark inferencedtypedevicearchmin_sqnrcompilerrrrnamemoder extra_informodeltorchaortypeoriginsmetricactualtargetrbenchmark_values target_value) benchmarkrr!r.jsonafileN) enumerateopenospathsplitextprintjsondumps)output_json_pathheadersrowivmapping_headersrecordfs r write_json_result_osscir<s2;3@wz1}O@($W-)(3'/+J7*95   $F+!{ $H-!0!: ;+H5 %F2 !!"23A67u=s Cq djj q) D C5A4 D CsB/B55 Cc6[U5VVs0sH up4XU_M nnn[RR5n[R "[ U5RS55R5n[RRU[RR5n[UR55n U S-n [[R"55SSSSUU SU SSUS US US US US S.S.USSS/S.USUS/USS.S. n [[R R#U5SS3S5n [%[&R("U 5U S9 SSS5 gs snnf!,(df  g=f)aI 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) zutf-8v3zdevvm local benchmarkz pytorch/aomainrrrrrrrrrrrrrr!r"r#r$) timestampschema_versionrrepo head_branchhead_sha workflow_id run_attemptjob_idr'rr!rr(r)r*N)r,datetimedatetodayhashlibsha256strencode hexdigestcombinetimeminintrAr-r.r/r0r1r2r3) r4r5r6r7r8r9rKsha_hash first_secondrFrHr:r;s r write_json_result_localrW>s2;3@wz1}O@ MM   !E~~c%j//89CCEH$$,,UHMM4E4EFLl,,./K 1_F%'"'$W-)(3'/+J7*95   $F+!{ $H-!0!: ;+H5 7 FD !!"23A67u=s Cq djj q) D CQAP D CsFF  F) rIrLr2r.r rRrrNrr<rWr r r rXs3   "s"#*L1*r