test.py

func_entry_float = ""
func_entry_double = ""
A100_float_bd = [13, 23]
A100_double_bd = [14, 23]
T4_float_bd = [13, 23]
T4_double_bd = [13, 23]

float_bd =  {"T4":T4_float_bd, "A100":A100_float_bd}
double_bd =  {"T4":T4_double_bd, "A100":A100_double_bd}

gpus = ["T4", "A100"]
spec = {"T4":75, "A100":80}
def helper_include(dtype, if_ft, if_err_inj, bd, gpu, spec):
    func_entry = f'''
template<> struct TurboFFT_Kernel_Entry<{dtype}, {if_ft}, {if_err_inj}, {spec}>
'''
    name = f"void (*turboFFTArr [26][3])({dtype} *, {dtype} *, {dtype} *, {dtype} *, int, int) ="
    func_entry += '{\n' + name + '{\n {NULL, NULL, NULL},\n'
    for i in range(1, 26):
        func_0 = f'''fft_radix_2<{dtype}, {i}, 0, {if_ft}, {if_err_inj}>'''
        func_1 = f'''fft_radix_2<{dtype}, {i}, 1, {if_ft}, {if_err_inj}>''' if i >= bd[0] else "NULL"
        func_2 = f'''fft_radix_2<{dtype}, {i}, 2, {if_ft}, {if_err_inj}>''' if i >= bd[1] else "NULL"
        func_entry += "{" + func_0 + ", " +  func_1 + ", " + func_2 + "},\n"
    func_entry += '''
};
};
'''
    print(func_entry)

for gpu in gpus:
    helper_include("float2", if_ft=0, if_err_inj=0, bd=float_bd[gpu], gpu=gpu, spec=spec[gpu])
    helper_include("float2", if_ft=1, if_err_inj=0, bd=float_bd[gpu], gpu=gpu, spec=spec[gpu])
    helper_include("float2", if_ft=1, if_err_inj=1, bd=float_bd[gpu], gpu=gpu, spec=spec[gpu])


    helper_include("double2", if_ft=0, if_err_inj=0, bd=double_bd[gpu], gpu=gpu, spec=spec[gpu])
    helper_include("double2", if_ft=1, if_err_inj=0, bd=double_bd[gpu], gpu=gpu, spec=spec[gpu])
    helper_include("double2", if_ft=1, if_err_inj=1, bd=double_bd[gpu], gpu=gpu, spec=spec[gpu])