[SOT] Extend SOT warmup support to new hardware (#3032)

* add new hardware

* add_sot_warmup4new_hardware

* fix conflict

* rm Optional
This commit is contained in:
Ryan
2025-07-29 22:45:20 +08:00
committed by GitHub
parent b2f9a42d87
commit 73cfe1fd37
7 changed files with 62 additions and 4 deletions

View File

@@ -70,9 +70,10 @@ class XpuWorker(WorkerBase):
def graph_optimize_and_warm_up_model(self) -> None:
"""
Optimizes the inference graph using the specified optimization options.
Perform the warm-up and the graph optimization
"""
logger.warn("XPU current could not graph optimize and warm up model")
if self.model_runner.graph_opt_level >= 1:
self.model_runner.sot_warmup()
def determine_available_memory(self) -> int:
"""