benchmark.py

"""
    This python script can be used to generate output (cuda program) files
    for all the input files present in INPUT_DIR.
"""

import os
import subprocess

INPUT_DIRS = ["benchmark/input/small/","benchmark/input/medium/","benchmark/input/large/","benchmark/input/extralarge/"]
OUTPUT_DIRS = ["benchmark/output/small/","benchmark/output/medium/","benchmark/output/large/","benchmark/output/extralarge/"]
SRC_DIR = "src/"

# Make executable from source files
_ = subprocess.run( 'make -C ' + SRC_DIR, shell = True, stdout=open(os.devnull, 'wb'))

# Run executable on input files
for INPUT_DIR,OUTPUT_DIR in zip(INPUT_DIRS,OUTPUT_DIRS):
    os.makedirs( OUTPUT_DIR, exist_ok = True)
    for root, dirs, files in os.walk(INPUT_DIR):
        for name in files:
            output_filename = name.split('.')[0] + '.cu'
            _ = subprocess.run( 'cd ' + SRC_DIR + ' && ./a.out ' + '../' + os.path.join(root, name)\
                 + ' ../' + OUTPUT_DIR + output_filename, shell = True )
            print("Generated " + OUTPUT_DIR + output_filename)

_ = subprocess.run( 'make clean -C ' + SRC_DIR, shell = True, stdout=open(os.devnull, 'wb'))