mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-14 02:57:45 +00:00
Add trait file for update_moving_average and save_mean_inv_std
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user