native top_p_sampling (#2901)

2026-04-23 00:17:25 +08:00 · 2025-07-22 14:09:59 +08:00
parent 0eedbdaee0
commit 2c6a9e887e
14 changed files with 93 additions and 7 deletions
@@ -173,6 +173,7 @@ class Sampler(nn.Layer):
            or current_platform.is_xpu()
            or current_platform.is_iluvatar()
            or current_platform.is_gcu()
+            or current_platform.is_dcu()
        ):
            self.forward = self.forward_cuda
        else: