apply review

apache · Feb 9, 2022 · 45f631c · 45f631c
1 parent ced7e26
commit 45f631c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/operator/nn/dnnl/dnnl_masked_softmax.cc b/src/operator/nn/dnnl/dnnl_masked_softmax.cc
@@ -133,6 +133,7 @@ void DNNLMaskedSoftmaxFwd::Execute(const Tensors& tensors,
   using namespace mxnet::op::mxnet_op;
   using namespace dnnl;
   /*
+   Three steps of masked softmax:
    1. out = input * [(mask - 1) * inf]
    2. out = softmax(out)
    3. out = out * mask

diff --git a/src/operator/nn/masked_softmax.cc b/src/operator/nn/masked_softmax.cc
@@ -21,13 +21,13 @@
  * \file masked_softmax.cc
  */
 
-#include "./softmax-inl.h"
-#include "../tensor/elemwise_unary_op.h"
-#include "../tensor/elemwise_binary_op.h"
-#include "../operator_common.h"
+#include "softmax-inl.h"
+#include "operator/tensor/elemwise_unary_op.h"
+#include "operator/tensor/elemwise_binary_op.h"
+#include "operator/operator_common.h"
 #if MXNET_USE_ONEDNN == 1
-#include "dnnl/dnnl_base-inl.h"
-#include "dnnl/dnnl_ops-inl.h"
+#include "operator/nn/dnnl/dnnl_base-inl.h"
+#include "operator/nn/dnnl/dnnl_ops-inl.h"
 #endif
 
 namespace mxnet {