PipelineStateVk: run legalization pass for HLSL shaders when remappin…

…g resources
DiligentGraphics · Jan 19, 2025 · 68e441e · 68e441e
1 parent 315c2ad
commit 68e441e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/Graphics/GraphicsEngineVulkan/src/PipelineStateVkImpl.cpp b/Graphics/GraphicsEngineVulkan/src/PipelineStateVkImpl.cpp
@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2024 Diligent Graphics LLC
+ *  Copyright 2019-2025 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -740,7 +740,12 @@ void PipelineStateVkImpl::RemapOrVerifyShaderResources(
                 //     SPIR-V module not valid: DecorateStringGOOGLE requires one of the following extensions: SPV_GOOGLE_decorate_string
                 // Optimizer also performs validation and may catch problems with the byte code.
                 // NB: SPIRV offsets become INVALID after this operation.
-                auto StrippedSPIRV = OptimizeSPIRV(SPIRV, SPV_ENV_MAX, SPIRV_OPTIMIZATION_FLAG_STRIP_REFLECTION);
+                SPIRV_OPTIMIZATION_FLAGS OptimizationFlags = SPIRV_OPTIMIZATION_FLAG_STRIP_REFLECTION;
+                if (pShaderResources->IsHLSLSource())
+                {
+                    OptimizationFlags |= SPIRV_OPTIMIZATION_FLAG_LEGALIZATION;
+                }
+                auto StrippedSPIRV = OptimizeSPIRV(SPIRV, SPV_ENV_MAX, OptimizationFlags);
                 if (!StrippedSPIRV.empty())
                     SPIRV = std::move(StrippedSPIRV);
                 else