diff --git a/include/ck_tile/ops/batchnorm/pipeline/batchnorm_fwd_traits.hpp b/include/ck_tile/ops/batchnorm/pipeline/batchnorm_fwd_traits.hpp new file mode 100644 index 0000000000..4556dfe4c1 --- /dev/null +++ b/include/ck_tile/ops/batchnorm/pipeline/batchnorm_fwd_traits.hpp @@ -0,0 +1,18 @@ +// SPDX-License-Identifier: MIT +// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved. + +#pragma once + +namespace ck_tile { + +// Traits for batchnorm forward pass configuration +// These are compile-time flags that create different kernel variants +template +struct BatchnormFwdTraits +{ + static constexpr bool kSaveMeanInvStd = kSaveMeanInvStd_; + static constexpr bool kUpdateMovingAverage = kUpdateMovingAverage_; +}; + +} // namespace ck_tile