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

Adapt for iluvatar gpu (#2684)

Browse Source
This commit is contained in:
liddk1121
2025-07-07 16:53:14 +08:00
committed by GitHub
parent 2579e8fea8
commit 1b54a2831e
50 changed files with 4485 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/model_executor/layers/utils.py
View File

@@ -33,6 +33,8 @@ if current_platform.is_cuda() and current_platform.available():
"Verify environment consistency between compilation and FastDeploy installation. "
"And ensure the Paddle version supports FastDeploy's custom operators"
)
if current_platform.is_iluvatar():
from fastdeploy.model_executor.ops.iluvatar import get_padding_offset
import re
from fastdeploy import envs
@@ -377,4 +379,4 @@ def create_and_set_parameter(layer: nn.Layer, name: str,
dtype=tensor.dtype,
default_initializer=paddle.nn.initializer.Constant(0),
))
getattr(layer, name).set_value(tensor)
getattr(layer, name).set_value(tensor)