[Cherry-Pick]Support finetuning the model saved on the MAC on the Li…

…nux (#34027) (#34154) 修复《jit.save在Mac系统上保存的模型，在Linux平台上无法对模型进行重训练》的问题。原始PR： #34027
PaddlePaddle · Jul 15, 2021 · 1d1ca0f · 1d1ca0f
1 parent 999c291
commit 1d1ca0f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/operators/matmul_op.cc b/paddle/fluid/operators/matmul_op.cc
@@ -232,7 +232,9 @@ class MatMulGradKernel : public framework::OpKernel<T> {
     int head_number = 1;
 #if defined(PADDLE_WITH_MKLML) && !defined(PADDLE_WITH_CUDA) && \
     !defined(PADDLE_WITH_HIP)
-    head_number = context.Attr<int>("head_number");
+    if (context.HasAttr("head_number")) {
+      head_number = context.Attr<int>("head_number");
+    }
 #endif
 
     if (head_number <= 1 && a.dims().size() == 3 && b.dims().size() <= 2) {