combine_DGL_MGG_8GPU_GCN.py

import csv
import os

# Read the first CSV file and get a column
with open('MGG_GCN_8GPU.csv', 'r') as file1:
    reader = csv.reader(file1)
    col0 = [row[0] for row in reader]

with open('dgl_pydirect_internal/gcn/1_dgl_gcn.csv', 'r') as file1:
    reader = csv.reader(file1)
    col1 = [float(row[1]) for row in reader]
    

# print(col0)
# print(col1)

# Read the second CSV file and get a column
with open('MGG_GCN_8GPU.csv', 'r') as file2:
    reader = csv.reader(file2)
    col2 = [float(row[1]) for row in reader]
# print(col2)

# Compute the ratio of the values in the two columns
ratio = [col1[i] / col2[i] for i in range(len(col1))]

# Write the result, along with the original two columns, to a third CSV file
with open('Fig_7_DGL_MGG_4GPU_GCN.csv', 'w', newline='') as result_file:
    writer = csv.writer(result_file)
    writer.writerow(['Dataset', 'DGL-8GPU(ms)', 'MGG-8GPU(ms)', 'Speedup (x)'])
    for i in range(len(col1)):
        writer.writerow([col0[i].rstrip("_beg_pos.bin"), col1[i], col2[i], "{:.3f}".format(ratio[i])])

print("please check the csv file: Fig_7_DGL_MGG_4GPU_GCN.csv")
# os.system("mv UVM_4GPU.csv csvs/")
# os.system("mv MGG_4GPU.csv csvs/")
os.system("mv *.err logs/")