apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[CI]【Hackathon 9th Sprint No.13】NO.13 功能模块 fastdeploy/model_executor/ops/triton_ops/triton_utils.py 单测补充 (#5035)

Browse Source 
* Add unit tests for triton_utils.py

* update name

* update

* update

* update
This commit is contained in:
Winters Montagne
2025-11-17 11:43:31 +08:00
committed by GitHub
parent 36216e62f0
commit 02c83d65db
2 changed files with 518 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/model_executor/ops/triton_ops/triton_utils.py
View File

@@ -710,7 +710,7 @@ class KernelInterface:
+ f""" -s"{address_hint} {value_hint} {const_args}" """
+ f""" -g "{lanuch_grid}" """
)
all_tune_config = list(self.tune_config)
all_tune_config = [{key: value} for key, value in self.tune_config.items()]
if len(all_tune_config) == 0:
# when user do not specify config, we use const_hint_dict as config.
all_tune_config = [const_hint_dict]