diff --git a/compiler/tests/delay_data_collection.py b/compiler/tests/delay_data_collection.py new file mode 100644 index 00000000..b5af25fc --- /dev/null +++ b/compiler/tests/delay_data_collection.py @@ -0,0 +1,137 @@ +#!/usr/bin/env python3 +""" +Run a regression test on various srams +""" + +import unittest +from testutils import header,openram_test +import sys,os +sys.path.append(os.path.join(sys.path[0],"..")) +import globals +from globals import OPTS +import debug +import csv +from sram import sram +from sram_config import sram_config + +MODEL_DIR = "model_data/" + +class data_collection(openram_test): + + def runTest(self): + self.init_data_gen() + + #Run on one size to initialize CSV writing (csv names come from return value). Strange, but it is okay for now. + sram_data = self.get_sram_data(1,16,1) + self.initialize_csv_file(sram_data) + + self.add_sram_data_to_csv(sram_data, 1, 16, 1) + + #Run openRAM for several size configurations + #word_size_list, num_word_list, words_per_row_list = self.get_sram_configs() + word_size_list, num_word_list, words_per_row_list = [4], [16], [1] #for quick testing. + for word_size in word_size_list: + for num_word in num_word_list: + for words_per_row in words_per_row_list: + #Unfortunately, init needs to be called everytime + self.init_data_gen() + sram_data = self.get_sram_data(word_size, num_word, words_per_row) + self.add_sram_data_to_csv(sram_data, word_size, num_word, words_per_row) + + self.close_files() + debug.info(1,"Data Generated") + globals.end_openram() + + def init_data_gen(self): + """Initialization for the data test to run""" + globals.init_openram("config_20_{0}".format(OPTS.tech_name)) + OPTS.spice_name="hspice" #Much faster than ngspice. + OPTS.trim_netlist = False + OPTS.netlist_only = True + OPTS.analytical_delay = False + # This is a hack to reload the characterizer __init__ with the spice version + from importlib import reload + import characterizer + reload(characterizer) + + def close_files(self): + """Closes all files stored in the file dict""" + for key,file in self.csv_files.items(): + file.close() + + def get_sram_configs(self): + """Generate lists of wordsizes, number of words, and column mux size (words per row) to be tested.""" + min_word_size = 1 + max_word_size = 16 + min_num_words_log2 = 4 + max_num_words_log2 = 8 + word_sizes = [i for i in range(min_word_size,max_word_size+1)] + num_words = [2**i for i in range(min_num_words_log2,max_num_words_log2+1)] + words_per_row = [1] + return word_sizes, num_words, words_per_row + + def add_sram_data_to_csv(self, sram_data, word_size, num_words, words_per_row): + """Writes data to its respective CSV file. There is a CSV for each measurement target (wordline, sense amp enable, and models)""" + sram_specs = [word_size,num_words,words_per_row] + for data_name, data_values in sram_data.items(): + self.csv_writers[data_name].writerow(sram_specs+sram_data[data_name]) + debug.info(2,"Data Added to CSV file.") + + def initialize_csv_file(self, sram_data): + """Opens a CSV file and writer for every data set being written (wl/sae measurements and model values)""" + #CSV File writing + header_dict = self.delay_obj.get_all_signal_names() + self.csv_files = {} + self.csv_writers = {} + for data_name, header_list in header_dict.items(): + self.csv_files[data_name] = open('{}data_{}.csv'.format(MODEL_DIR,data_name), 'w') + fields = ('word_size', 'num_words', 'words_per_row', *header_list) + self.csv_writers[data_name] = csv.writer(self.csv_files[data_name], lineterminator = '\n') + self.csv_writers[data_name].writerow(fields) + + def get_sram_data(self, word_size, num_words, words_per_row): + """Generates the SRAM based on input configuration and returns the data.""" + from characterizer import model_check + c = sram_config(word_size=word_size, + num_words=num_words, + num_banks=1) + #minimum 16 rows. Most sizes below 16*16 will try to automatically use less rows unless enforced. + #if word_size*num_words < 256: + c.words_per_row=words_per_row #Force no column mux until incorporated into analytical delay. + + debug.info(1, "Getting data for {} bit, {} words SRAM with 1 bank".format(word_size, num_words)) + s = sram(c, name="sram_{}ws_{}words".format(word_size, num_words)) + + tempspice = OPTS.openram_temp + "temp.sp" + s.sp_write(tempspice) + + corner = (OPTS.process_corners[0], OPTS.supply_voltages[0], OPTS.temperatures[0]) + self.delay_obj = model_check(s.s, tempspice, corner) + + import tech + #Only 1 at a time + probe_address = "1" * s.s.addr_size + probe_data = s.s.word_size - 1 + loads = [tech.spice["msflop_in_cap"]*4] + slews = [tech.spice["rise_time"]*2] + + sram_data = self.delay_obj.analyze(probe_address,probe_data,slews,loads) + return sram_data + + def remove_lists_from_dict(self, dict): + """Check all the values in the dict and replaces the list items with its first value.""" + #This is useful because the tests performed here only generate 1 value but a list + #with 1 item makes writing it to a csv later harder. + for key in dict.keys(): + if type(dict[key]) is list: + if len(dict[key]) > 0: + dict[key] = dict[key][0] + else: + del dict[key] + +# instantiate a copdsay of the class to actually run the test +if __name__ == "__main__": + (OPTS, args) = globals.parse_args() + del sys.argv[1:] + header(__file__, OPTS.tech_name) + unittest.main() diff --git a/compiler/tests/model_data/data_sae_measures.csv b/compiler/tests/model_data/data_sae_measures.csv new file mode 100644 index 00000000..29b9ee07 --- /dev/null +++ b/compiler/tests/model_data/data_sae_measures.csv @@ -0,0 +1,3 @@ +word_size,num_words,words_per_row,Xsram.Xcontrol0.Xand2_rbl_in.zb_int,Xsram.Xcontrol0.rbl_in,Xsram.Xcontrol0.Xreplica_bitline.Xdelay_chain.dout_1,Xsram.Xcontrol0.Xreplica_bitline.delayed_en,Xsram.Xcontrol0.pre_s_en,Xsram.Xcontrol0.Xbuf_s_en.zb_int,Xsram.s_en0 +1,16,1,0.1073407,0.010667,0.1317698,0.0910598,0.04802989999999999,0.1304522,0.0164942 +4,16,1,0.10635089999999998,0.010598499999999999,0.1315219,0.0911604,0.048601,0.12349109999999999,0.05874360000000001 diff --git a/compiler/tests/model_data/data_sae_model.csv b/compiler/tests/model_data/data_sae_model.csv new file mode 100644 index 00000000..25a56d9e --- /dev/null +++ b/compiler/tests/model_data/data_sae_model.csv @@ -0,0 +1,3 @@ +word_size,num_words,words_per_row,Xsram.Xcontrol0.Xand2_rbl_in.zb_int,Xsram.Xcontrol0.rbl_in,Xsram.Xcontrol0.Xreplica_bitline.Xdelay_chain.dout_1,Xsram.Xcontrol0.Xreplica_bitline.delayed_en,Xsram.Xcontrol0.pre_s_en,Xsram.Xcontrol0.Xbuf_s_en.zb_int,Xsram.s_en0 +1,16,1,5.0,0.75,4.5,5.5,2.5,4.5,1.09375 +4,16,1,5.0,0.75,4.5,5.5,2.5,4.5,1.09375 diff --git a/compiler/tests/model_data/data_wl_measures.csv b/compiler/tests/model_data/data_wl_measures.csv new file mode 100644 index 00000000..95aa4044 --- /dev/null +++ b/compiler/tests/model_data/data_wl_measures.csv @@ -0,0 +1,3 @@ +word_size,num_words,words_per_row,Xsram.Xcontrol0.Xbuf_wl_en.zb_int,Xsram.wl_en0,Xsram.Xbank0.Xwordline_driver0.wl_bar_15,Xsram.Xbank0.wl_15 +1,16,1,0.1257435,0.0364958,0.051171100000000004,0.020759299999999998 +4,16,1,0.124175,0.03654059999999999,0.0507837,0.0433494 diff --git a/compiler/tests/model_data/data_wl_model.csv b/compiler/tests/model_data/data_wl_model.csv new file mode 100644 index 00000000..ae601f49 --- /dev/null +++ b/compiler/tests/model_data/data_wl_model.csv @@ -0,0 +1,3 @@ +word_size,num_words,words_per_row,Xsram.Xcontrol0.Xbuf_wl_en.zb_int,Xsram.wl_en0,Xsram.Xbank0.Xwordline_driver0.wl_bar_15,Xsram.Xbank0.wl_15 +1,16,1,1.5,21.833333333333332,2.0,1.1666666666666665 +4,16,1,2.5,11.166666666666666,2.0,3.1666666666666665