apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

polish code with new pre-commit rule (#2923)

Browse Source
This commit is contained in:
Zero Rains
2025-07-19 23:19:27 +08:00
committed by GitHub
parent b8676d71a8
commit 25698d56d1
424 changed files with 14307 additions and 13518 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastdeploy/platforms/utils.py
View File

@@ -19,6 +19,7 @@
import numpy as np
import paddle
def convert_to_npu_dequant_scale(deq_scale):
"""
Convert dequantization scale for NPU.
@@ -39,8 +40,5 @@ def convert_to_npu_dequant_scale(deq_scale):
if not paddle.is_compiled_with_custom_device("npu"):
return deq_scale
arr = deq_scale.numpy()
new_deq_scale = np.stack(
[arr.reshape(-1, 1),
np.zeros_like(arr).reshape(-1, 1)], axis=-1).reshape(-1)
return paddle.to_tensor(
np.frombuffer(new_deq_scale.tobytes(), dtype=np.int64))
new_deq_scale = np.stack([arr.reshape(-1, 1), np.zeros_like(arr).reshape(-1, 1)], axis=-1).reshape(-1)
return paddle.to_tensor(np.frombuffer(new_deq_scale.tobytes(), dtype=np.int64))