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

Commit

Permalink
fix sanity
Browse files Browse the repository at this point in the history
  • Loading branch information
bgawrych committed Oct 26, 2021
1 parent bf0b9a6 commit 53d8d53
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
2 changes: 1 addition & 1 deletion src/operator/subgraph/dnnl/dnnl_batch_dot.cc
Original file line number Diff line number Diff line change
Expand Up @@ -173,4 +173,4 @@ NNVM_REGISTER_OP(_sg_onednn_batch_dot)
} // namespace op
} // namespace mxnet

#endif // if MXNET_USE_ONEDNN == 1
#endif // if MXNET_USE_ONEDNN == 1
6 changes: 1 addition & 5 deletions src/operator/subgraph/dnnl/dnnl_batch_dot_property.h
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,6 @@ namespace op {

class SgDNNLBatchDotSelector : public SubgraphSelector {
public:
explicit SgDNNLBatchDotSelector() {}

bool Select(const nnvm::Node& n) override {
return n.op() && n.op()->name == "batch_dot";
}
Expand All @@ -51,8 +49,6 @@ class SgDNNLBatchDotSelector : public SubgraphSelector {

class SgDNNLBatchDotProperty : public SubgraphProperty {
public:
SgDNNLBatchDotProperty() {}

static SubgraphPropertyPtr Create() {
static const std::string& name = "DNNL Batch Dot optimization pass";
auto property = std::make_shared<SgDNNLBatchDotProperty>();
Expand Down Expand Up @@ -100,4 +96,4 @@ class SgDNNLBatchDotProperty : public SubgraphProperty {
} // namespace mxnet

#endif // if MXNET_USE_ONEDNN == 1
#endif // MXNET_OPERATOR_SUBGRAPH_DNNL_DNNL_BATCH_DOT_PROPERTY_H_
#endif // MXNET_OPERATOR_SUBGRAPH_DNNL_DNNL_BATCH_DOT_PROPERTY_H_

0 comments on commit 53d8d53

Please sign in to comment.