From 4dbc74a953ae1b34ce64a4eaef3aa7feb73663b9 Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 13 Feb 2023 10:38:14 -0800 Subject: Add Loop Unrolling Pass. (#2644) Co-authored-by: Yong He --- source/slang/slang-check-modifier.cpp | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) (limited to 'source/slang/slang-check-modifier.cpp') diff --git a/source/slang/slang-check-modifier.cpp b/source/slang/slang-check-modifier.cpp index 9f3e79978..520d85971 100644 --- a/source/slang/slang-check-modifier.cpp +++ b/source/slang/slang-check-modifier.cpp @@ -507,15 +507,29 @@ namespace Slang // as 1 arg if nothing is specified) SLANG_ASSERT(attr->args.getCount() == 1); } + else if (auto forceUnrollAttr = as(attr)) + { + if (forceUnrollAttr->args.getCount() < 1) + { + getSink()->diagnose(attr, Diagnostics::notEnoughArguments, attr->args.getCount(), 1); + } + auto cint = checkConstantIntVal(attr->args[0]); + if (cint) + forceUnrollAttr->maxIterations = (int32_t)cint->value; + } else if (auto maxItersAttrs = as(attr)) { - if (auto cint = checkConstantIntVal(attr->args[0])) + if (attr->args.getCount() < 1) { - maxItersAttrs->value = (int32_t) cint->value; + getSink()->diagnose(attr, Diagnostics::notEnoughArguments, attr->args.getCount(), 1); } else { - getSink()->diagnose(attr, Diagnostics::notEnoughArguments, attr->args.getCount(), 1); + auto cint = checkConstantIntVal(attr->args[0]); + if (cint) + { + maxItersAttrs->value = (int32_t) cint->value; + } } } else if (auto userDefAttr = as(attr)) -- cgit v1.2.3