diff --git a/lib/Conversion/TorchToTosa/TorchToTosa.cpp b/lib/Conversion/TorchToTosa/TorchToTosa.cpp index ace593bf4f0a..dbe13300385c 100644 --- a/lib/Conversion/TorchToTosa/TorchToTosa.cpp +++ b/lib/Conversion/TorchToTosa/TorchToTosa.cpp @@ -3447,29 +3447,32 @@ LogicalResult ConvertAtenOp::matchAndRewrite( std::multiplies()); Value half = tosa::getConstTensor(rewriter, op, - SmallVector(numElem, 0.5), + SmallVector(numElem, 0.5f), selfShape, selfElemTy) .value(); Value one = tosa::getConstTensor(rewriter, op, - SmallVector(numElem, 1.0), + SmallVector(numElem, 1.0f), selfShape, selfElemTy) .value(); Value three = tosa::getConstTensor(rewriter, op, - SmallVector(numElem, 3.0), + SmallVector(numElem, 3.0f), selfShape, selfElemTy) .value(); // 0.044715 - Value magicNumber = tosa::getConstTensor( - rewriter, op, SmallVector(numElem, 0.044715), - selfShape, selfElemTy) - .value(); + Value magicNumber = + tosa::getConstTensor(rewriter, op, + SmallVector(numElem, 0.044715f), + selfShape, selfElemTy) + .value(); // From header: M_2_PI = 2 / pi - Value twoOverPi = tosa::getConstTensor( - rewriter, op, SmallVector(numElem, M_2_PI), - selfShape, selfElemTy) - .value(); + Value twoOverPi = + tosa::getConstTensor( + rewriter, op, + SmallVector(numElem, static_cast(M_2_PI)), selfShape, + selfElemTy) + .value(); // 0.5 * x auto halfInput = rewriter.create(op->getLoc(), resultType,