diff --git a/source/backend/cpu/CPULinSpace.cpp b/source/backend/cpu/CPULinSpace.cpp index b6c8b47d..0be6d55e 100644 --- a/source/backend/cpu/CPULinSpace.cpp +++ b/source/backend/cpu/CPULinSpace.cpp @@ -17,7 +17,6 @@ ErrorCode CPULinSpace::onExecute(const std::vector& inputs, const std:: const float stop = inputs[1]->host()[0]; const int num = inputs[2]->host()[0]; MNN_ASSERT(num > 0); - const float step = (stop - start) / (num - 1); float* outputData = outputs[0]->host(); @@ -35,6 +34,7 @@ ErrorCode CPULinSpace::onExecute(const std::vector& inputs, const std:: // make sure that start with the first and end with the last. outputData[0] = start; outputData[num - 1] = stop; + const float step = (stop - start) / (num - 1); for (int i = 1; i < num - 1; ++i) { outputData[i] = start + i * step; }