17 lines
680 B
C++
17 lines
680 B
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "NNEHlslShadersBatchNormalizationCS.h"
|
|
#include "NNE.h"
|
|
|
|
namespace UE::NNEHlslShaders::Internal
|
|
{
|
|
void TBatchNormalizationCS::ModifyCompilationEnvironment(const FGlobalShaderPermutationParameters& InParameters, FShaderCompilerEnvironment& OutEnvironment)
|
|
{
|
|
FGlobalShader::ModifyCompilationEnvironment(InParameters, OutEnvironment);
|
|
|
|
OutEnvironment.SetDefine(TEXT("THREADGROUP_SIZE_X"), FBatchNormalizationConstants::NUM_GROUP_THREADS);
|
|
}
|
|
|
|
IMPLEMENT_GLOBAL_SHADER(TBatchNormalizationCS, "/NNEHlslShaders/NNEHlslShadersBatchNormalization.usf", "BatchNormalization", SF_Compute);
|
|
} // UE::NNEHlslShaders::Internal
|