trace_gen_wrapper.py

import math
import dram_trace as dram
import sram_traffic_os as sram
import sram_traffic_ws as sram_ws
import sram_traffic_is as sram_is

def gen_all_traces(
        array_h = 4,
        array_w = 4,
        ifmap_h = 7, ifmap_w = 7,
        filt_h  = 3, filt_w = 3,
        num_channels = 3,
        strides = 1, num_filt = 8,

        data_flow = 'os',

        word_size_bytes = 1,
        filter_sram_size = 64, ifmap_sram_size= 64, ofmap_sram_size = 64,

        filt_base = 1000000, ifmap_base=0, ofmap_base = 2000000,
        sram_read_trace_file = "sram_read.csv",
        sram_write_trace_file = "sram_write.csv",

        dram_filter_trace_file = "dram_filter_read.csv",
        dram_ifmap_trace_file = "dram_ifmap_read.csv",
        dram_ofmap_trace_file = "dram_ofmap_write.csv"
    ):

    sram_cycles = 0
    util        = 0

    print("Generating traces and bw numbers")
    if data_flow == 'os':
        sram_cycles, util = \
            sram.sram_traffic(
                dimension_rows= array_h,
                dimension_cols= array_w,
                ifmap_h=ifmap_h, ifmap_w=ifmap_w,
                filt_h=filt_h, filt_w=filt_w,
                num_channels=num_channels,
                strides=strides, num_filt=num_filt,
                filt_base=filt_base, ifmap_base=ifmap_base,
                ofmap_base = ofmap_base,
                sram_read_trace_file=sram_read_trace_file,
                sram_write_trace_file=sram_write_trace_file
            )
    elif data_flow == 'ws':
        sram_cycles, util = \
            sram_ws.sram_traffic(
                dimension_rows = array_h,
                dimension_cols = array_w,
                ifmap_h = ifmap_h, ifmap_w = ifmap_w,
                filt_h = filt_h, filt_w = filt_w,
                num_channels = num_channels,
                strides = strides, num_filt = num_filt,
                ofmap_base = ofmap_base, filt_base = filt_base, ifmap_base = ifmap_base,
                sram_read_trace_file = sram_read_trace_file,
                sram_write_trace_file = sram_write_trace_file
            )
    elif data_flow == 'is':
        sram_cycles, util = \
            sram_is.sram_traffic(
                dimension_rows = array_h,
                dimension_cols = array_w,
                ifmap_h = ifmap_h, ifmap_w = ifmap_w,
                filt_h = filt_h, filt_w = filt_w,
                num_channels = num_channels,
                strides = strides, num_filt = num_filt,
                ofmap_base = ofmap_base, filt_base = filt_base, ifmap_base = ifmap_base,
                sram_read_trace_file = sram_read_trace_file,
                sram_write_trace_file = sram_write_trace_file
            )

    #print("Generating DRAM traffic")
    dram.dram_trace_read_v2(
        sram_sz=ifmap_sram_size,
        word_sz_bytes=word_size_bytes,
        min_addr=ifmap_base, max_addr=filt_base,
        sram_trace_file=sram_read_trace_file,
        dram_trace_file=dram_ifmap_trace_file,
    )

    dram.dram_trace_read_v2(
        sram_sz= filter_sram_size,
        word_sz_bytes= word_size_bytes,
        min_addr=filt_base, max_addr=ofmap_base,
        sram_trace_file= sram_read_trace_file,
        dram_trace_file= dram_filter_trace_file,
    )

    dram.dram_trace_write(
        ofmap_sram_size= ofmap_sram_size,
        data_width_bytes= word_size_bytes,
        sram_write_trace_file= sram_write_trace_file,
        dram_write_trace_file= dram_ofmap_trace_file
    )

    print("Average utilization : \t"  + str(util) + " %")
    print("Cycles for compute  : \t"  + str(sram_cycles) + " cycles")
    bw_numbers, detailed_log  = gen_bw_numbers(dram_ifmap_trace_file, dram_filter_trace_file,
                                 dram_ofmap_trace_file, sram_write_trace_file,
                                 sram_read_trace_file)
                                 #array_h, array_w)

    return bw_numbers, detailed_log, util, sram_cycles


def gen_max_bw_numbers( dram_ifmap_trace_file, dram_filter_trace_file,
                    dram_ofmap_trace_file, sram_write_trace_file, sram_read_trace_file
                    ):

    max_dram_activation_bw = 0
    num_bytes = 0
    max_dram_act_clk = ""
    f = open(dram_ifmap_trace_file, 'r')

    for row in f:
        clk = row.split(',')[0]
        num_bytes = len(row.split(',')) - 2
        
        
        if max_dram_activation_bw < num_bytes:
            max_dram_activation_bw = num_bytes
            max_dram_act_clk = clk
    f.close()

    max_dram_filter_bw = 0
    num_bytes = 0
    max_dram_filt_clk = ""
    f = open(dram_filter_trace_file, 'r')

    for row in f:
        clk = row.split(',')[0]
        num_bytes = len(row.split(',')) - 2

        if max_dram_filter_bw < num_bytes:
            max_dram_filter_bw = num_bytes
            max_dram_filt_clk = clk

    f.close()

    max_dram_ofmap_bw = 0
    num_bytes = 0
    max_dram_ofmap_clk = ""
    f = open(dram_ofmap_trace_file, 'r')

    for row in f:
        clk = row.split(',')[0]
        num_bytes = len(row.split(',')) - 2

        if max_dram_ofmap_bw < num_bytes:
            max_dram_ofmap_bw = num_bytes
            max_dram_ofmap_clk = clk

    f.close()
    
    max_sram_ofmap_bw = 0
    num_bytes = 0
    f = open(sram_write_trace_file, 'r')

    for row in f:
        num_bytes = len(row.split(',')) - 2

        if max_sram_ofmap_bw < num_bytes:
            max_sram_ofmap_bw = num_bytes

    f.close()

    max_sram_read_bw = 0
    num_bytes = 0
    f = open(sram_read_trace_file, 'r')

    for row in f:
        num_bytes = len(row.split(',')) - 2

        if max_sram_read_bw < num_bytes:
            max_sram_read_bw = num_bytes

    f.close()

    #print("DRAM IFMAP Read BW, DRAM Filter Read BW, DRAM OFMAP Write BW, SRAM OFMAP Write BW")
    log  = str(max_dram_activation_bw) + ",\t" + str(max_dram_filter_bw) + ",\t" 
    log += str(max_dram_ofmap_bw) + ",\t" + str(max_sram_read_bw) + ",\t"
    log += str(max_sram_ofmap_bw)  + ","
    # Anand: Enable the following for debug print
    #log += str(max_dram_act_clk) + ",\t" + str(max_dram_filt_clk) + ",\t"
    #log += str(max_dram_ofmap_clk) + ","
    #print(log)
    return log


def gen_bw_numbers( dram_ifmap_trace_file, dram_filter_trace_file,
                    dram_ofmap_trace_file, sram_write_trace_file, 
                    sram_read_trace_file
                    #sram_read_trace_file,
                    #array_h, array_w        # These are needed for utilization calculation
                    ):

    min_clk = 100000
    max_clk = -1
    detailed_log = ""

    num_dram_activation_bytes = 0
    f = open(dram_ifmap_trace_file, 'r')
    start_clk = 0
    first = True

    for row in f:
        num_dram_activation_bytes += len(row.split(',')) - 2
        
        elems = row.strip().split(',')
        clk = float(elems[0])

        if first:
            first = False
            start_clk = clk

        if clk < min_clk:
            min_clk = clk

    stop_clk = clk
    detailed_log += str(start_clk) + ",\t" + str(stop_clk) + ",\t" + str(num_dram_activation_bytes) + ",\t"
    f.close()

    num_dram_filter_bytes = 0
    f = open(dram_filter_trace_file, 'r')
    first = True

    for row in f:
        num_dram_filter_bytes += len(row.split(',')) - 2

        elems = row.strip().split(',')
        clk = float(elems[0])

        if first:
            first = False
            start_clk = clk

        if clk < min_clk:
            min_clk = clk

    stop_clk = clk
    detailed_log += str(start_clk) + ",\t" + str(stop_clk) + ",\t" + str(num_dram_filter_bytes) + ",\t"
    f.close()

    num_dram_ofmap_bytes = 0
    f = open(dram_ofmap_trace_file, 'r')
    first = True

    for row in f:
        num_dram_ofmap_bytes += len(row.split(',')) - 2

        elems = row.strip().split(',')
        clk = float(elems[0])

        if first:
            first = False
            start_clk = clk

    stop_clk = clk
    detailed_log += str(start_clk) + ",\t" + str(stop_clk) + ",\t" + str(num_dram_ofmap_bytes) + ",\t"
    f.close()
    if clk > max_clk:
        max_clk = clk
    

    num_sram_ofmap_bytes = 0
    f = open(sram_write_trace_file, 'r')
    first = True

    for row in f:
        num_sram_ofmap_bytes += len(row.split(',')) - 2
        elems = row.strip().split(',')
        clk = float(elems[0])

        if first:
            first = False
            start_clk = clk

    stop_clk = clk
    detailed_log += str(start_clk) + ",\t" + str(stop_clk) + ",\t" + str(num_sram_ofmap_bytes) + ",\t"
    f.close()
    if clk > max_clk:
        max_clk = clk
    
    num_sram_read_bytes = 0
    total_util = 0
    #print("Opening " + sram_trace_file)
    f = open(sram_read_trace_file, 'r')
    first = True

    for row in f:
        #num_sram_read_bytes += len(row.split(',')) - 2
        elems = row.strip().split(',')
        clk = float(elems[0])

        if first:
            first = False
            start_clk = clk

        #util, valid_bytes = parse_sram_read_data(elems[1:-1], array_h, array_w)
        valid_bytes = parse_sram_read_data(elems[1:])
        num_sram_read_bytes += valid_bytes
        #total_util += util
        #print("Total Util " + str(total_util) + ", util " + str(util))

    stop_clk = clk
    detailed_log += str(start_clk) + ",\t" + str(stop_clk) + ",\t" + str(num_sram_read_bytes) + ",\t"
    f.close()
    sram_clk = clk
    if clk > max_clk:
        max_clk = clk

    delta_clk = max_clk - min_clk

    dram_activation_bw  = num_dram_activation_bytes / delta_clk
    dram_filter_bw      = num_dram_filter_bytes / delta_clk
    dram_ofmap_bw       = num_dram_ofmap_bytes / delta_clk
    sram_ofmap_bw       = num_sram_ofmap_bytes / delta_clk
    sram_read_bw        = num_sram_read_bytes / delta_clk
    #print("total_util: " + str(total_util) + ", sram_clk: " + str(sram_clk))
    #avg_util            = total_util / sram_clk * 100

    units = " Bytes/cycle"
    print("DRAM IFMAP Read BW  : \t" + str(dram_activation_bw) + units)
    print("DRAM Filter Read BW : \t" + str(dram_filter_bw) + units)
    print("DRAM OFMAP Write BW : \t" + str(dram_ofmap_bw) + units)
    #print("Average utilization : \t"  + str(avg_util) + " %")
    #print("SRAM OFMAP Write BW, Min clk, Max clk")
    
    log = str(dram_activation_bw) + ",\t" + str(dram_filter_bw) + ",\t" + str(dram_ofmap_bw) + ",\t" + str(sram_read_bw) + ",\t" + str(sram_ofmap_bw) + ","
    # Anand: Enable the following line for debug
    #log += str(min_clk) + ",\t" + str(max_clk) + ","
    #print(log)
    #return log, avg_util
    return log, detailed_log


#def parse_sram_read_data(elems, array_h, array_w):
def parse_sram_read_data(elems):
    #half = int(len(elems) /2)
    #nz_row = 0
    #nz_col = 0
    data = 0

    for i in range(len(elems)):
        e = elems[i]
        if e != ' ':
            data += 1
            #if i < half:
            #if i < array_h:
            #    nz_row += 1
            #else:
            #    nz_col += 1

    #util = (nz_row * nz_col) / (half * half)
    #util = (nz_row * nz_col) / (array_h * array_w)
    #data = nz_row + nz_col
    
    #return util, data
    return data


def test():
    test_fc1_24x24 = [27, 37, 512, 27, 37, 512, 1, 24, 1]
    test_yolo_tiny_conv1_24x24 = [418, 418, 3, 3, 3, 16, 1, 24, 1]
    test_mdnet_conv1_24x24 = [107, 107, 3, 7, 7, 96, 2, 24, 1]

    #param = test_fc1_24x24
    #param = test_yolo_tiny_conv1_24x24
    param = test_mdnet_conv1_24x24

    # The parameters for 1st layer of yolo_tiny
    ifmap_h = param[0]
    ifmap_w = param[1]
    num_channels = param[2]

    filt_h = param[3]
    filt_w = param[4]
    num_filters = param[5] #16

    strides = param[6]

    # Model parameters
    dimensions = param[7] #32 #16
    word_sz = param[8]

    filter_sram_size = 1 * 1024
    ifmap_sram_size = 1 * 1024
    ofmap_sram_size = 1 * 1024

    filter_base = 1000000
    ifmap_base = 0
    ofmap_base = 2000000

    # Trace files
    sram_read_trace = "test_sram_read.csv"
    sram_write_trace  = "test_sram_write.csv"

    dram_filter_read_trace = "test_dram_filt_read.csv"
    dram_ifmap_read_trace  = "test_dram_ifamp_read.csv"
    dram_write_trace = "test_dram_write.csv"

    gen_all_traces(
        array_h = dimensions,
        array_w = dimensions,
        ifmap_h= ifmap_h, ifmap_w= ifmap_w, num_channels=num_channels,
        filt_h= filt_h, filt_w= filt_w, num_filt= num_filters,
        strides= strides,

        filter_sram_size= filter_sram_size, ifmap_sram_size= ifmap_sram_size, ofmap_sram_size= ofmap_sram_size,
        word_size_bytes= word_sz, filt_base= filter_base, ifmap_base= ifmap_base,

        sram_read_trace_file= sram_read_trace, sram_write_trace_file= sram_write_trace,

        dram_filter_trace_file= dram_filter_read_trace,
        dram_ifmap_trace_file= dram_ifmap_read_trace,
        dram_ofmap_trace_file= dram_write_trace
    )


if __name__ == "__main__":
    test()