From 393185196ed65a9eeaf9502edbf3dcce87337d81 Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 22 Aug 2022 09:43:05 -0700 Subject: Support compile-time constant int val in the form of polynomials. (#2372) Co-authored-by: Yong He --- source/slang/slang-lower-to-ir.cpp | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'source/slang/slang-lower-to-ir.cpp') diff --git a/source/slang/slang-lower-to-ir.cpp b/source/slang/slang-lower-to-ir.cpp index 1285afca8..3d4f0df5d 100644 --- a/source/slang/slang-lower-to-ir.cpp +++ b/source/slang/slang-lower-to-ir.cpp @@ -1287,6 +1287,27 @@ struct ValLoweringVisitor : ValVisitorastBuilder, val->declRef))); } + LoweredValInfo visitPolynomialIntVal(PolynomialIntVal* val) + { + auto irBuilder = getBuilder(); + auto constTerm = irBuilder->getIntValue(irBuilder->getIntType(), val->constantTerm); + auto resultVal = constTerm; + for (auto term : val->terms) + { + auto termVal = irBuilder->getIntValue(irBuilder->getIntType(), term->constFactor); + for (auto factor : term->paramFactors) + { + auto factorVal = lowerVal(context, factor->param).val; + for (IntegerLiteralValue i = 0; i < factor->power; i++) + { + termVal = irBuilder->emitMul(factorVal->getDataType(), termVal, factorVal); + } + } + resultVal = irBuilder->emitAdd(termVal->getDataType(), resultVal, termVal); + } + return LoweredValInfo::simple(resultVal); + } + LoweredValInfo visitDeclaredSubtypeWitness(DeclaredSubtypeWitness* val) { return emitDeclRef(context, val->declRef, -- cgit v1.2.3