diff --git a/include/aidge/backend/cpu/operator/BitShiftImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/BitShiftImpl_forward_kernels.hpp index b8b809b69bd479ee122b9981d17817b91c68410a..760187351fb6ba476d60eda4370de6553655520c 100644 --- a/include/aidge/backend/cpu/operator/BitShiftImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/BitShiftImpl_forward_kernels.hpp @@ -46,6 +46,7 @@ void BitShiftImpl_cpu_forward_kernel( std::size_t idx1 = getFlattenedIndex(input1Dims, indexes); std::size_t idx2 = getFlattenedIndex(input2Dims, indexes); if(direction == BitShift_Op::BitShiftDirection::right) + { output[oIndex]= input_1[idx1] >> input_2[idx2]; }