30 lines
1.3 KiB
C++
30 lines
1.3 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "NNEHlslShadersCumSumCS.h"
|
|
#include "DataDrivenShaderPlatformInfo.h"
|
|
#include "NNE.h"
|
|
#include "RenderGraphBuilder.h"
|
|
|
|
namespace UE::NNEHlslShaders::Internal
|
|
{
|
|
void TInitCumSumCS::ModifyCompilationEnvironment(const FGlobalShaderPermutationParameters& InParameters, FShaderCompilerEnvironment& OutEnvironment)
|
|
{
|
|
FGlobalShader::ModifyCompilationEnvironment(InParameters, OutEnvironment);
|
|
OutEnvironment.SetDefine(TEXT("INIT_THREADGROUP_SIZE"), FCumSumConstants::INIT_THREADGROUP_SIZE);
|
|
OutEnvironment.SetDefine(TEXT("INIT_SHADER"), 1);
|
|
}
|
|
|
|
void TCumSumCS::ModifyCompilationEnvironment(const FGlobalShaderPermutationParameters& InParameters, FShaderCompilerEnvironment& OutEnvironment)
|
|
{
|
|
FGlobalShader::ModifyCompilationEnvironment(InParameters, OutEnvironment);
|
|
|
|
OutEnvironment.SetDefine(TEXT("THREADGROUP_SIZE"), FCumSumConstants::THREADGROUP_SIZE);
|
|
OutEnvironment.SetDefine(TEXT("VALUES_PER_THREAD"), FCumSumConstants::VALUES_PER_THREAD);
|
|
|
|
FPermutationDomain PermutationVector(InParameters.PermutationId);
|
|
}
|
|
|
|
IMPLEMENT_GLOBAL_SHADER(TInitCumSumCS, "/NNEHlslShaders/NNEHlslShadersCumSum.usf", "InitCumSum", SF_Compute);
|
|
IMPLEMENT_GLOBAL_SHADER(TCumSumCS, "/NNEHlslShaders/NNEHlslShadersCumSum.usf", "CumSum", SF_Compute);
|
|
} // UE::NNEHlslShaders::Internal
|