Adapt for iluvatar gpu (#2684)

2025-10-05 00:33:03 +08:00 · 2025-07-07 16:53:14 +08:00
parent 2579e8fea8
commit 1b54a2831e
50 changed files with 4485 additions and 80 deletions
--- a/fastdeploy/model_executor/layers/activation.py
+++ b/fastdeploy/model_executor/layers/activation.py
@@ -63,7 +63,8 @@ class SiluAndMul(nn.Layer):
        """
        super().__init__()

-        if current_platform.is_cuda() or current_platform.is_xpu():
+        if current_platform.is_cuda() or current_platform.is_xpu(
+        ) or current_platform.is_iluvatar():
            self.forward = self.forward_cuda
        else:
            raise NotImplementedError