[XPU] bug fix for xpu_parallel_matmul tensor parallel output. (Paddle…

…Paddle#9297)
lvdongyi · Oct 23, 2024 · 32e3ccc · 32e3ccc
1 parent 3412f50
commit 32e3ccc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/transformers/tensor_parallel_utils.py b/paddlenlp/transformers/tensor_parallel_utils.py
@@ -44,7 +44,7 @@ def parallel_matmul(lm_output, logit_weights, tensor_parallel_output=True, train
             logits = xpu_parallel_matmul(
                 lm_output,
                 logit_weights,
-                tensor_parallel_output,
+                tensor_parallel_output=tensor_parallel_output,
                 training=training,
             )
             return logits