From c4fbbaaa5533b2689f4f3c42f0b3d5f3bb1376e5 Mon Sep 17 00:00:00 2001
From: Mohsen Saffari <mohsen.saffari@amd.com>
Date: Fri, 28 Nov 2025 10:33:58 +0000
Subject: [PATCH] Add trait file for update_moving_average and
 save_mean_inv_std

---
 .../pipeline/batchnorm_fwd_traits.hpp          | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 include/ck_tile/ops/batchnorm/pipeline/batchnorm_fwd_traits.hpp

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 <bool kSaveMeanInvStd_,
+          bool kUpdateMovingAverage_>
+struct BatchnormFwdTraits
+{
+    static constexpr bool kSaveMeanInvStd = kSaveMeanInvStd_;
+    static constexpr bool kUpdateMovingAverage = kUpdateMovingAverage_;
+};
+
+} // namespace ck_tile