[XPU] Enable qwen3.5 on XPU (#21668)

Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
This commit is contained in:
Xia Weiwen
2026-05-18 14:59:19 +08:00
committed by GitHub
co-authored by Ma Mingfei
parent 1f9eda4ea1
commit 8d5ed330cc
14 changed files with 757 additions and 13 deletions
+11 -7
View File
@@ -174,7 +174,9 @@ def stop_profile(
if save_trace:
if profiler is not None:
if trace_filename:
_save_profile_trace_results(profiler, trace_filename)
_save_profile_trace_results(
profiler, profile_activities, trace_filename
)
stage_desc = f"for {stage}" if stage else ""
rank_print(
f"torch profiler chrome trace {stage_desc} saved to {trace_filename}"
@@ -597,15 +599,17 @@ def _create_torch_profiler_filename(
return os.path.join(output_dir, filename)
def _save_profile_trace_results(profiler, filename):
def _save_profile_trace_results(profiler, profile_activities, filename):
parent_dir = os.path.dirname(os.path.abspath(filename))
os.makedirs(parent_dir, exist_ok=True)
profiler.export_chrome_trace(filename)
print(
profiler.key_averages(group_by_input_shape=True).table(
sort_by="self_cpu_time_total"
)
)
if "GPU" in profile_activities:
sort_by = "self_cuda_time_total"
elif "XPU" in profile_activities:
sort_by = "self_xpu_time_total"
else:
sort_by = "self_cpu_time_total"
print(profiler.key_averages(group_by_input_shape=True).table(sort_by=sort_by))
def correctness_test(