monitor_efa_aws.py

monitor_efa_aws.py

from glob import glob
import time
import datetime

def get_read_bytes():
    return sum([int(open(f"{p}/ports/1/hw_counters/rdma_read_bytes", "r").read().strip()) for p in glob("/sys/class/infiniband/*")])
from os.path import expanduser
home = expanduser("~")


with open(f"{home}/speed_measure", "a") as f:
    while 1:
        now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        start = get_read_bytes()
        time.sleep(1)
        end = get_read_bytes()
        s = now+"\t"+str((end-start)/(10**9)) + "\t" + "GB/s" + "\n"
        f.write(s)
        print(s, end='')
        f.flush()