diff options
| author | Yong He <yonghe@outlook.com> | 2024-03-11 14:42:14 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-11 14:42:14 -0700 |
| commit | 1bbcf25af514a9ae24f7006747177f2d1b3b7c0d (patch) | |
| tree | f42c17d32040d033742e741548e7b73ff24a5e92 /source/slang/slang-reflection-api.cpp | |
| parent | 25a7d51445e64253beca5c4f70ddd52f40226b1d (diff) | |
Link-time specialization fixes. (#3734)
* Fix method synthesis logic for static differentiable methods.
* Support link-time constants in thread group size reflection.
Diffstat (limited to 'source/slang/slang-reflection-api.cpp')
| -rw-r--r-- | source/slang/slang-reflection-api.cpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/source/slang/slang-reflection-api.cpp b/source/slang/slang-reflection-api.cpp index 7af3ce0a3..d91dd5858 100644 --- a/source/slang/slang-reflection-api.cpp +++ b/source/slang/slang-reflection-api.cpp @@ -2811,12 +2811,18 @@ SLANG_API void spReflectionEntryPoint_getComputeThreadGroupSize( auto numThreadsAttribute = entryPointFunc.getDecl()->findModifier<NumThreadsAttribute>(); if (numThreadsAttribute) { - if (auto cint = as<ConstantIntVal>(numThreadsAttribute->x)) + if (auto cint = entryPointLayout->program->tryFoldIntVal(numThreadsAttribute->x)) sizeAlongAxis[0] = (SlangUInt)cint->getValue(); - if (auto cint = as<ConstantIntVal>(numThreadsAttribute->y)) + else if (numThreadsAttribute->x) + sizeAlongAxis[0] = 0; + if (auto cint = entryPointLayout->program->tryFoldIntVal(numThreadsAttribute->y)) sizeAlongAxis[1] = (SlangUInt)cint->getValue(); - if (auto cint = as<ConstantIntVal>(numThreadsAttribute->z)) + else if (numThreadsAttribute->y) + sizeAlongAxis[1] = 0; + if (auto cint = entryPointLayout->program->tryFoldIntVal(numThreadsAttribute->z)) sizeAlongAxis[2] = (SlangUInt)cint->getValue(); + else if (numThreadsAttribute->z) + sizeAlongAxis[2] = 0; } // |