benchmark.py

import os
import sys
import time
import matplotlib.pyplot as plt

def buildDict(logs_type):
    type_dict = {}
    for logs in logs_type:
        arr = logs.split(":")
        wordLen = int(arr[0])
        time = float(arr[-1])

        if wordLen not in type_dict:
            type_dict[wordLen] = []
        type_dict[wordLen].append(time)
    return type_dict
    


numIterations = [512, 1024, 10000, 20000, 50000, 100000, 150000, 200000, 250000, 300000, 350000, 370000]
numLengths = [10]

for length in numLengths:
    for iteration in numIterations:
        #print("running: ", length, " ", iteration, "\n")
        openmp = "./Openmp.o " + str(length) + " " + str(iteration) + " " + str(8)
        basic = "./Basic.o " + str(length) + " " + str(iteration)  
        cuda = "./Cuda.o " + str(length) + " " + str(iteration) + " " + str(100) + " " + str(500)
        #print(basic)
        os.system(cuda)
        os.system(openmp)
        os.system(basic)

#Read  text files and plot here
#Basic plot
#Multiple lines: each line represent word length, for each iteration how much time


f_basic = open("./Times/basic_times.txt")
f_openmp = open("./Times/openmp_times.txt")
f_cuda = open("./Times/cuda_times.txt")

basic = f_basic.read().split('\n')
openmp = f_openmp.read().split('\n')
cuda = f_cuda.read().split('\n')

f_basic.close()
f_openmp.close()
f_cuda.close()

basic = basic[:-1]
openmp = openmp[:-1]
cuda = cuda[:-1]

basic_dict = buildDict(basic)
openmp_dict = buildDict(openmp)
cuda_dict = buildDict(cuda)


x = numIterations
for wordLen in numLengths:
    y_basic = basic_dict[wordLen]
    y_openmp = openmp_dict[wordLen]
    y_cuda = cuda_dict[wordLen]

    plt.figure()
    plt.plot(x, y_basic, marker="x", label = "Basic")
    plt.plot(x, y_openmp, marker="x", label = "OpenMP")
    plt.plot(x, y_cuda, marker="x", label="CUDA")
    plt.legend()
    plt.title("Word Length: " + str(wordLen))
    plt.xlabel("Number of Iterations")
    plt.ylabel("Time in ms")
plt.show()