mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 08:37:06 +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:
@@ -123,7 +123,8 @@ class GcuWorker(WorkerBase):
|
||||
"""
|
||||
# 1. Warm up model
|
||||
# NOTE(gongshaotian): may be not need warm_up at this place
|
||||
|
||||
if self.model_runner.graph_opt_level >= 1:
|
||||
self.model_runner.sot_warmup()
|
||||
# 2. Triger cuda grpah capture
|
||||
self.model_runner.capture_model()
|
||||
|
||||
|
Reference in New Issue
Block a user