Add Loop Unrolling Pass. (#2644)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 17 insertions, 3 deletions

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<ForceUnrollAttribute>(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<MaxItersAttribute>(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<UserDefinedAttribute>(attr))