Skip to content

Conversation

@tensor-tang
Copy link
Contributor

@tensor-tang tensor-tang commented Nov 16, 2017

check trainer_count when use_mkldnn

finish the PR in #5671

@tensor-tang tensor-tang requested a review from luotao1 November 16, 2017 15:36

if (FLAGS_trainer_count > 1) {
CHECK(!FLAGS_use_mkldnn) << "MKLDNN only need 1 trainer";
}
Copy link
Contributor

Choose a reason for hiding this comment

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

这个check要调换一下顺序么:

if (FLAGS_use_mkldnn) {
     CHECK_EQ(FLAGS_trainer_count, 1) << "MKLDNN only need 1 trainer";
}

毕竟用use_mkldnn的情况少,就不用每次做check了。

Copy link
Contributor Author

Choose a reason for hiding this comment

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

好的

@tensor-tang
Copy link
Contributor Author

done

Copy link
Contributor

@luotao1 luotao1 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 ba9f252 into PaddlePaddle:develop Nov 17, 2017
@tensor-tang tensor-tang deleted the check branch November 17, 2017 03:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants