Function write_merge_file_name

dnn/scripts/cutlass_generator/gen_list.py:40–44

(f, gen_op, gen_type, split_number)

# Write down a list of merged filenames
def write_merge_file_name(f, gen_op, gen_type, split_number):
    for i in range(0, split_number):
        f.write('    "{}_{}_{}.cu",\n'.format(gen_op, gen_type, i))
    if gen_op != "gemv":
        f.write('    "all_{}_{}_operations.cu",\n'.format(gen_op, gen_type))


if __name__ == "__main__":