From bbd1e1786401bb88c34802b987d4da72e2364503 Mon Sep 17 00:00:00 2001 From: Yong He Date: Wed, 1 Feb 2023 14:18:57 -0800 Subject: Support `out` parameters in backward differentiation. (#2619) * Support `out` parameters in backward differentiation. * Fixes. * Fix cleanup. --------- Co-authored-by: Yong He --- source/slang/slang-ir-init-local-var.cpp | 34 ++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 source/slang/slang-ir-init-local-var.cpp (limited to 'source/slang/slang-ir-init-local-var.cpp') diff --git a/source/slang/slang-ir-init-local-var.cpp b/source/slang/slang-ir-init-local-var.cpp new file mode 100644 index 000000000..4b28db268 --- /dev/null +++ b/source/slang/slang-ir-init-local-var.cpp @@ -0,0 +1,34 @@ +// slang-ir-init-local-var.cpp +#include "slang-ir-init-local-var.h" +#include "slang-ir.h" +#include "slang-ir-insts.h" + +namespace Slang +{ + +void initializeLocalVariables(SharedIRBuilder* sharedBuilder, IRGlobalValueWithCode* func) +{ + IRBuilder builder(sharedBuilder); + for (auto block : func->getBlocks()) + { + for (auto inst : block->getChildren()) + { + if (inst->getOp() == kIROp_Var) + { + auto firstUse = inst->firstUse; + bool initialized = + (firstUse && firstUse->getUser()->getOp() == kIROp_Store && + firstUse->getUser()->getParent() == inst->getParent()); + if (initialized) + continue; + builder.setInsertAfter(inst); + builder.emitStore( + inst, + builder.emitDefaultConstruct( + as(inst->getFullType())->getValueType())); + } + } + } +} + +} // namespace Slang -- cgit v1.2.3