convolution_specialization.hpp Source File

convolution_specialization.hpp Source File#

Composable Kernel: convolution_specialization.hpp Source File

// SPDX-License-Identifier: MIT
// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
 
#pragma once
 
#include <string>
 
namespace ck_tile {
 
enum struct ConvolutionSpecialization
{
    Default,
    Filter1x1Pad0,
    Filter1x1Stride1Pad0,
    Filter3x3,
};
 
CK_TILE_HOST std::string getConvSpecializationString(const ConvolutionSpecialization& s)
{
    switch(s)
    {
    case ConvolutionSpecialization::Default: return "Default";
    case ConvolutionSpecialization::Filter1x1Pad0: return "Filter1x1Pad0";
    case ConvolutionSpecialization::Filter1x1Stride1Pad0: return "Filter1x1Stride1Pad0";
    case ConvolutionSpecialization::Filter3x3: return "Filter3x3";
    default: return "Unrecognized specialization!";
    }
}
 
} // namespace ck_tile