From 32e3cccc41d32f695fb9eaddd1394c1b33a9b1e6 Mon Sep 17 00:00:00 2001
From: Yuang Liu <liuyuang@baidu.com>
Date: Tue, 22 Oct 2024 16:25:42 +0800
Subject: [PATCH] [XPU] bug fix for xpu_parallel_matmul tensor parallel output.
 (#9297)

---
 paddlenlp/transformers/tensor_parallel_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlenlp/transformers/tensor_parallel_utils.py b/paddlenlp/transformers/tensor_parallel_utils.py
index 267c2846aa27..ea68e41207c5 100644
--- 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