Skip to content

Commit 5f675df

Browse files
committed
fix KGroupedGemmXPUFunction bug
1 parent dff7725 commit 5f675df

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

paddle/phi/kernels/legacy/xpu/batched_gemm_xpu_utils.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,11 @@ static void KGroupedGemmXPUFunction(const DenseTensor &a,
178178
scale_y,
179179
scale_x_mode,
180180
scale_y_mode);
181+
} else {
182+
xpu::constant(xpu_ctx,
183+
reinterpret_cast<XPUType *>(output_data),
184+
input_hidden_size * output_hidden_size,
185+
(XPUType)0);
181186
}
182187
a_data += expert_bs * input_hidden_size;
183188
b_data += expert_bs * output_hidden_size;

0 commit comments

Comments
 (0)