Update attn_gemm.h

PaddlePaddle · Sep 21, 2023 · 841e553 · 841e553
1 parent 8abb807
commit 841e553
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/operators/fused/attn_gemm.h b/paddle/fluid/operators/fused/attn_gemm.h
@@ -255,10 +255,10 @@ class AttnMatMul {
       gpuStream_t stream = dev_ctx_.stream();
       if (support_case_1 || support_case_2) {
         phi::SumKernel<T>(
-            dev_ctx, *d_output, {0, 1}, d_output->dtype(), false, d_bias);
+            dev_ctx_, *d_output, {0, 1}, d_output->dtype(), false, d_bias);
       } else if (support_case_3 || support_case_4) {
         phi::SumKernel<T>(
-            dev_ctx, *d_output, {0, 2}, d_output->dtype(), false, d_bias);
+            dev_ctx_, *d_output, {0, 2}, d_output->dtype(), false, d_bias);
       } else {
         PADDLE_THROW(platform::errors::InvalidArgument(
             "Only support reduce when the input dims are [0,1,2,3,4] and "