| 12345678910111213141516171819202122 |
- # 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
|
