Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
add comments to false statement in macro
Browse files Browse the repository at this point in the history
  • Loading branch information
Bartlomiej Gawrych committed Nov 10, 2021
1 parent b4c335d commit 215b79b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/operator/tensor/matrix_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,8 @@ bool ReshapeStorageType(const nnvm::NodeAttrs& attrs,
std::vector<int>* out_attrs) {
CHECK_EQ(in_attrs->size(), 1U);
CHECK_EQ(out_attrs->size(), 1U);
return DNNLStorageType(attrs, dev_mask, true, dispatch_mode, in_attrs, out_attrs);
return DNNLStorageType(
attrs, dev_mask, /*support_dnnl*/ true, dispatch_mode, in_attrs, out_attrs);
}
#endif

Expand Down Expand Up @@ -944,7 +945,7 @@ static void StackForwardEx(const nnvm::NodeAttrs& attrs,
}

if (SupportDNNLStack(inputs)) {
DNNL_OPCHECK_INIT(false, outputs.size(), inputs, outputs);
DNNL_OPCHECK_INIT(/*is backward*/ false, outputs.size(), inputs, outputs);
DNNLRun(DNNLStackForward, attrs, op_ctx, inputs, req, outputs);
DNNL_OPCHECK_RUN(StackOpForward<cpu>, attrs, op_ctx, inputs, req, outputs);
} else {
Expand Down

0 comments on commit 215b79b

Please sign in to comment.