mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 13:11:25 +00:00
0be0288f58
* Update copyright messages. Copyright messages should no longer include a year. This PR updates all 38 source files to the new format. * Switch to (C) from unicode copyright symbol. The unicodein comments was causing compilation errors.
39 lines
1.3 KiB
C++
39 lines
1.3 KiB
C++
// Copyright (C) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
#include <concepts>
|
|
#include <type_traits>
|
|
|
|
#include "ck_tile/builder/conv_factory.hpp"
|
|
#include "ck_tile/builder/versions.hpp"
|
|
|
|
namespace ck_tile::builder {
|
|
|
|
/**
|
|
* @brief Top-level builder for creating convolution kernel instances.
|
|
*
|
|
* This struct serves as the main entry point for generating a convolution kernel.
|
|
* It uses a factory pattern based on the provided signature, algorithm, and version
|
|
* to construct the appropriate kernel instance.
|
|
*
|
|
* @tparam SIGNATURE The convolution signature, which describes the mathematical functionality of
|
|
* the algorithm (e.g., data types, layouts, direction).
|
|
* @tparam ALGORITHM The specific convolution algorithm to be used for the implementation.
|
|
* @tparam VERSION The version of the builder implementation.
|
|
*/
|
|
template <ConvSignatureDescriptor auto SIGNATURE,
|
|
ConvAlgorithmDescriptor auto ALGORITHM,
|
|
StringLiteral VERSION = LATEST_API_VERSION>
|
|
requires SupportedVersion<VERSION> && ValidConvSignature<SIGNATURE>
|
|
struct ConvBuilder
|
|
{
|
|
static constexpr auto kVersion = VERSION;
|
|
using Factory = ConvFactory<SIGNATURE, ALGORITHM, VERSION>;
|
|
// Output: The kernel class.
|
|
using Instance = Factory::Instance;
|
|
};
|
|
|
|
} // namespace ck_tile::builder
|