diff options
Diffstat (limited to 'source/slang/slang-check-modifier.cpp')
| -rw-r--r-- | source/slang/slang-check-modifier.cpp | 20 |
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)) |