mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-30 03:22:05 +08:00
[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:
@@ -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:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user