-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
src/operator/nn/activation-inl.h
Outdated
@@ -199,13 +199,8 @@ void ActivationGradCompute(const nnvm::NodeAttrs& attrs, | |||
const std::vector<OpReqType>& req, | |||
const std::vector<TBlob>& outputs) { | |||
const ActivationParam& param = nnvm::get<ActivationParam>(attrs.parsed); | |||
#if (MXNET_USE_CUDNN == 1 || MXNET_USE_MKLDNN == 1) | |||
bool relu = param.act_type == activation::kReLU; | |||
const bool relu = param.act_type == activation::kReLU; | |||
CHECK_EQ(inputs.size(), relu ? 2U : 3U); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This code was added in #11827 . Maybe @samskalicky can help here and say whether it's needed
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I fixed all this stuff in #13409 just keeping the CI pipeline here until @marcoabreu green lights.
@mxnet-label-bot add [pr-awaiting-review] |
Please wait with merging this PR until the Jenkinsfile refactoring has been completed |
@mxnet-label-bot update [pr-awaiting-testing] |
@larroy ping, is this still needed? thanks! |
@roywei is needed. @marcoabreu pls review. |
19a730e
to
0518e16
Compare
0481230
to
37291f5
Compare
Ready to merge @marcoabreu |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Are the mshadow submodule changes needed?
@lebeg no, fixed. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me
Description
We were not running C++ test compiled without GPU and without MKL, which had test failures, such as #13409
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.