Adapt for iluvatar gpu (#2684)

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

View File

@@ -20,6 +20,7 @@ import numpy as np
import paddle
from paddle import nn
from paddle.incubate.nn.functional import fused_layer_norm, fused_rms_norm
from fastdeploy.platforms import current_platform
from fastdeploy.config import FDConfig
@@ -265,6 +266,18 @@ class LayerNorm(nn.Layer):
The `residual_output` is the result of applying the normalization and possibly other
operations (like linear transformation) on the `residual_input`.
"""
if current_platform.is_iluvatar():
if self.ln_weight is None and self.ln_bias is None:
out = x
if self.linear_bias is not None:
out += self.linear_bias
if residual_input is not None:
out += residual_input
return out, out
else:
return out
else:
raise NotImplementedError("Iluvatar does not support yet!")
norm_out = self.norm_func(
x,