Skip to content
Snippets Groups Projects

[Fix] Seg fault in Div forward file

Merged Maxence Naud requested to merge bugfix-6-divimpl-forward into dev
1 file
+ 4
10
Compare changes
  • Side-by-side
  • Inline
+ 4
10
@@ -91,16 +91,10 @@ void Aidge::DivImpl_cpu::forward() {
@@ -91,16 +91,10 @@ void Aidge::DivImpl_cpu::forward() {
std::size_t contiguousIdx = nbDims - 1;
std::size_t contiguousIdx = nbDims - 1;
for (; contiguousIdx+1 > 0; --contiguousIdx) {
for (; contiguousIdx+1 > 0; --contiguousIdx) {
if (dims0[contiguousIdx] != dims1[contiguousIdx]) {
if (dims0[contiguousIdx] != dims1[contiguousIdx]) {
if (contiguousIdx == (nbDims -1)) {
if (contiguousIdx == (nbDims -1)) { // last dimensions of one of the input Tensor are of size 1
if (dims0[contiguousIdx] == 1) {
const std::vector<std::size_t>& dims = (dims0[contiguousIdx] == 1) ? dims0 : dims1;
while ((dims0[contiguousIdx] == 1) && (contiguousIdx+1 > 0)) {
while ((contiguousIdx+1 > 0) && (dims[contiguousIdx] == 1)) {
--contiguousIdx;
--contiguousIdx;
}
}
else {
while ((dims1[contiguousIdx] == 1) && (contiguousIdx+1 > 0)) {
--contiguousIdx;
}
}
}
}
}
break;
break;
Loading