Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add FLAGS_use_mkldnn to global control use_mkldnn #11319

Merged
merged 3 commits into from
Jun 11, 2018
Merged

add FLAGS_use_mkldnn to global control use_mkldnn #11319

merged 3 commits into from
Jun 11, 2018

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].

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
Development

Successfully merging this pull request may close these issues.

Problem about use_mkldnn flag
2 participants