We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c1be08e commit be2eadbCopy full SHA for be2eadb
backends/mlu/kernels/funcs/reduce_op.h
@@ -38,7 +38,7 @@ void MLUReduceOp(const Context& dev_ctx,
38
const auto& input_dim_size = x.dims().size();
39
std::vector<int> reduce_dims;
40
VLOG(3) << "ReduceOp keep_dim " << keep_dim;
41
- if (!keep_dim && dims.size() == 0) reduce_all = true;
+ if (dims.size() == 0) reduce_all = true;
42
if (reduce_all) {
43
for (size_t i = 0; i < input_dims.size(); i++) {
44
reduce_dims.push_back(static_cast<int>(i));
0 commit comments