# counter for tracking how many kernels have been generated generated_kernel_count = 0 generated_cpp_vec_kernel_count = 0 num_bytes_accessed = 0 nodes_num_elem = [] # counters for tracking fusions ir_nodes_pre_fusion = 0 # reset all counters def reset(): global generated_kernel_count global generated_cpp_vec_kernel_count global num_bytes_accessed, nodes_num_elem global ir_nodes_pre_fusion generated_kernel_count = 0 generated_cpp_vec_kernel_count = 0 num_bytes_accessed = 0 nodes_num_elem.clear() ir_nodes_pre_fusion = 0