U ‰d¶ã@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ú@/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/numa_benchmark.pyÚ build_nets(  ÿ   þrc CsLt ¡rt ¡dkst‚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 IsNUMAEnabledZGetNumNUMANodesÚAssertionErrorrZ 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