We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 999c291 commit 1d1ca0fCopy full SHA for 1d1ca0f
paddle/fluid/operators/matmul_op.cc
@@ -232,7 +232,9 @@ class MatMulGradKernel : public framework::OpKernel<T> {
232
int head_number = 1;
233
#if defined(PADDLE_WITH_MKLML) && !defined(PADDLE_WITH_CUDA) && \
234
!defined(PADDLE_WITH_HIP)
235
- head_number = context.Attr<int>("head_number");
+ if (context.HasAttr("head_number")) {
236
+ head_number = context.Attr<int>("head_number");
237
+ }
238
#endif
239
240
if (head_number <= 1 && a.dims().size() == 3 && b.dims().size() <= 2) {
0 commit comments