[Diffusion] Fix and validate diffusion skills benchmarking/profiling workflow (#20528)

This commit is contained in:
Xiaoyu Zhang
2026-03-13 21:11:37 +08:00
committed by GitHub
parent b1246c50f8
commit be7a0311a0
10 changed files with 318 additions and 78 deletions
@@ -25,7 +25,9 @@ def load_engine_model():
json_files = glob.glob(os.path.join(os.path.dirname(__file__) or ".", "*.json"))
for fname in json_files:
with open(fname, encoding="utf-8") as f:
engine_model.update(json.load(f))
file_engine_model = json.load(f)
for engine, models in file_engine_model.items():
engine_model.setdefault(engine, {}).update(models)
return engine_model
@@ -38,7 +40,6 @@ class GPUTrace2Graph:
import pandas as pd # avoid importing till needed
self.pd = pd
self.pd.options.mode.copy_on_write = True
# helper functions for generating trace->summary csvs
def gen_nonoverlapped_sum_from_gputrace(self, in_file, out_file):