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

Commit 4e2fadb

Browse files
author
Bartlomiej Gawrych
committed
Fix windows GPU issue
1 parent 54eaf7e commit 4e2fadb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/operator/numpy/np_broadcast_reduce_op_value.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,12 +32,12 @@
3232
#include "../tvmop/op_module.h"
3333
#endif // MXNET_USE_TVM_OP
3434

35+
#include "np_broadcast_reduce_op.h"
36+
3537
#if MXNET_USE_ONEDNN
3638
#include "../nn/dnnl/dnnl_reduce-inl.h"
3739
#endif // MXNET_USE_ONEDNN
3840

39-
#include "np_broadcast_reduce_op.h"
40-
4141
namespace mxnet {
4242
namespace op {
4343

src/operator/tensor/broadcast_reduce_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1640,7 +1640,7 @@ void LpNormCompute(const nnvm::NodeAttrs& attrs,
16401640
}
16411641
#else
16421642
const std::string& red = param.ord == 1 ? "red::sum{}" : "red::nrm2{}";
1643-
const std::string& op = param.ord == 1 ? "abs" : "identity";
1643+
const std::string& op = param.ord == 1 ? "abs" : "identity";
16441644
ReduceAxesRTCComputeImpl(ctx, inputs, req, outputs, small, red, nullptr, false, op);
16451645
#endif
16461646
}

0 commit comments

Comments
 (0)