Adapt for iluvatar gpu (#2684)

2025-12-24 13:28:13 +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/sample/sampler.py
+++ b/fastdeploy/model_executor/layers/sample/sampler.py
@@ -170,7 +170,8 @@ class Sampler(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()