Skip to content

Conversation

@luotao1
Copy link
Contributor

@luotao1 luotao1 commented Jun 8, 2018

fix #10765

  • for python script, when GLOG_logtostderr=1 GLOG_v=3 FLAGS_use_mkldnn=true make test ARGS='-R test_image_classification -V':
...
I0608 18:07:09.210391 20222 executor.cc:385] use_mkldnn=True
...
290: I0608 18:43:14.423079  1224 operator.cc:638] Transform Variable conv2d_26.tmp_1 from data_type[float]:data_layout[NCHW]:place[CPUPlace]:library_type[PLAIN] to data_type[float]:data_layout[MKLDNNLAYOUT]:place[CPUPlace]:library_type[MKLDNN]
290: I0608 18:43:14.423085  1224 scope.cc:53] Create variable conv2d_26.tmp_1
290: I0608 18:43:14.423094  1224 operator.cc:638] Transform Variable batch_norm_26.tmp_2@GRAD from data_type[float]:data_layout[NCHW]:place[CPUPlace]:library_type[PLAIN] to data_type[float]:data_layout[MKLDNNLAYOUT]:place[CPUPlace]:library_type[MKLDNN]
  • for c++ script, when GLOG_logtostderr=1 GLOG_v=3 FLAGS_use_mkldnn=true make test ARGS='-R test_inference_image_classification -V':
...
53: I0608 19:08:13.280025 18482 executor.cc:322] CPUPlace Op(conv2d), inputs:{Filter[conv2d_65.w_0[64, 64, 3, 3]({})], Input[batch_norm_64.tmp_3[1, 64, 8, 8]({})]}, outputs:{Output[conv2d_65.tmp_0[1, 64, 8, 8]({})]}.
53: I0608 19:08:13.280045 18482 operator.cc:612] expected_kernel_key:data_type[float]:data_layout[MKLDNNLAYOUT]:place[CPUPlace]:library_type[MKLDNN]
... 

Thus, for both python and C++ script, we can find operator.cc use library_type[MKLDNN].

@luotao1 luotao1 requested review from Superjomn and tensor-tang June 8, 2018 11:29
Copy link
Contributor

@Superjomn Superjomn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@luotao1 luotao1 merged commit 69b5a62 into PaddlePaddle:develop Jun 11, 2018
@luotao1 luotao1 deleted the mkldnn branch June 11, 2018 04:39
@luotao1 luotao1 added the Intel label Jun 11, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Problem about use_mkldnn flag

2 participants