(p)
| 434 | repeat=5) |
| 435 | |
| 436 | def trace_handler(p): |
| 437 | output = p.key_averages().table(sort_by="self_cpu_time_total", row_limit=20) |
| 438 | # print(output) |
| 439 | p.export_chrome_trace(f"{RESULT_DIR}/{p.step_num}.json") |
| 440 | |
| 441 | for _ in range(10): |
| 442 | model(**input_dict) # compiled_model(**input_dict) to get inductor model profiling |
nothing calls this directly
no outgoing calls
no test coverage detected