a ==ic¶ã@sdddlmZmZddlmZddlZdZdZdZdZ dd „Z d d „Z e d kr`e  d dg¡e ƒdS)é)ÚcoreÚ workspace)Ú caffe2_pb2Niièi@é0cCsºt |d¡}d| ¡_t ¡}tj|_d|_t t ƒD]&}|j g|dt |ƒt t g|dq6t |¡}d| ¡_|r|d|_t t ƒD],}|j|dt |ƒ|dt |ƒ|dq„||fS) NÚ_initZasync_schedulingrz /input_blob_)ÚshapeÚ device_optionéz /output_blob_)r)rZNetZProtoÚtyperZ DeviceOptionZCPUZ device_typeZ numa_node_idÚrangeÚ NUM_REPLICASZ XavierFillÚstrÚ SHAPE_LENZCopy)Znet_nameZ cross_socketZinit_netZnuma_device_optionZ replica_idÚnet©rúm/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/numa_benchmark.pyÚ build_nets&  ÿ   þrc CsLt ¡rt ¡dksJ‚tddƒ\}}tddƒ\}}t |¡t | ¡¡t |¡t | ¡¡t |¡t |¡tdƒD]Æ}t ¡}t | ¡t ¡t ¡|}t d|ƒdt t t t |t }t d |¡ƒt ¡}t | ¡t ¡t ¡|}t d |ƒdt t t t |t }t d  |¡ƒt d  ||¡ƒq€dS) NéÚ single_netFÚ cross_netTézSingle socket time:zSingle socket BW: {} GB/szCross socket time:zCross socket BW: {} GB/szSingle BW / Cross BW: {})rZ IsNUMAEnabledZGetNumNUMANodesrZ CreateNetZRunNetÚNamer ÚtimeÚNUM_ITERÚprintrr ÚGBÚformat) Z single_initrZ cross_initrÚ_ÚtÚdtZ single_bwZcross_bwrrrÚmain$s.         r Ú__main__Zcaffe2z--caffe2_cpu_numa_enabled=1)Z caffe2.pythonrrZ caffe2.protorrrrrr rr Ú__name__Z GlobalInitrrrrÚs