Add trait file for update_moving_average and save_mean_inv_std

This commit is contained in:
Mohsen Saffari
2025-11-28 10:33:58 +00:00
parent b732595d9f
commit c4fbbaaa55

View File

@@ -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