From 0754abee603c1afa7803a444124acc9d268d2f0a Mon Sep 17 00:00:00 2001 From: Yong He Date: Tue, 12 Nov 2024 21:01:47 -0800 Subject: Push buffer load to end of access chain. (#5544) * Push buffer load to end of access chain. * Update test. * Fix. * Fix. * Fix. * Make more robust. * Fix. --- source/slang/slang-emit.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'source/slang/slang-emit.cpp') diff --git a/source/slang/slang-emit.cpp b/source/slang/slang-emit.cpp index 1950f251c..05bb12ecc 100644 --- a/source/slang/slang-emit.cpp +++ b/source/slang/slang-emit.cpp @@ -30,6 +30,7 @@ #include "slang-ir-com-interface.h" #include "slang-ir-composite-reg-to-mem.h" #include "slang-ir-dce.h" +#include "slang-ir-defer-buffer-load.h" #include "slang-ir-defunctionalization.h" #include "slang-ir-diff-call.h" #include "slang-ir-dll-export.h" @@ -951,6 +952,11 @@ Result linkAndOptimizeIR( // Inline calls to any functions marked with [__unsafeInlineEarly] or [ForceInline]. performForceInlining(irModule); + // Push `structuredBufferLoad` to the end of access chain to avoid loading unnecessary data. + if (isKhronosTarget(targetRequest) || isMetalTarget(targetRequest) || + isWGPUTarget(targetRequest)) + deferBufferLoad(irModule); + // Specialization can introduce dead code that could trip // up downstream passes like type legalization, so we // will run a DCE pass to clean up after the specialization. -- cgit v1.2.3