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:
@@ -434,7 +434,7 @@ class GraphOptimizationConfig:
|
||||
- With dyncmic graph backend: ...
|
||||
- With static grpah backend: WIP
|
||||
"""
|
||||
self.sot_warmup_sizes: Optional[list[int]] = []
|
||||
self.sot_warmup_sizes: list[int] = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 16, 32, 64, 128]
|
||||
""" Number of warmup runs for SOT warmup. """
|
||||
self.use_cudagraph: bool = False
|
||||
"""Sizes to capture cudagraph.
|
||||
|
Reference in New Issue
Block a user