Skip to content
Snippets Groups Projects
Commit 452a65e6 authored by Maxence Naud's avatar Maxence Naud Committed by Maxence Naud
Browse files

fix 'And' kernel in case of equal dimensions for inputs

parent 981290cd
No related branches found
No related tags found
2 merge requests!118v0.4.0,!40Resolve "Arithmetic Operator optimization"
...@@ -62,7 +62,7 @@ void EqualImpl_cpu_forward_kernel(std::vector<std::size_t> dims0, ...@@ -62,7 +62,7 @@ void EqualImpl_cpu_forward_kernel(std::vector<std::size_t> dims0,
const std::size_t input0_contiguous_size = std::accumulate(dims0.cbegin(), dims0.cend(), std::size_t(1), std::multiplies<std::size_t>()); const std::size_t input0_contiguous_size = std::accumulate(dims0.cbegin(), dims0.cend(), std::size_t(1), std::multiplies<std::size_t>());
for (std::size_t i = 0; i < input0_contiguous_size; ++i) for (std::size_t i = 0; i < input0_contiguous_size; ++i)
{ {
output[i] = static_cast<O>(input_0[i] + input_1[i]); output[i] = static_cast<O>(input_0[i] == input_1[i]);
} }
return; return;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment