mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-11 09:40:51 +00:00
Fix naming style of all signature enums.
Some of these enum values were CamelCase when they should have been CONSTANT_CASE.
This commit is contained in:
@@ -10,8 +10,8 @@ namespace ck_tile::builder {
|
||||
// Layouts for grouped convolutions.
|
||||
enum class GroupConvLayout
|
||||
{
|
||||
CHANNELS_LAST, // Channels-last NHWGC_GKYXC_NHWGK
|
||||
CHANNELS_FIRST // Channels-first NGCHW_GKCYX_NGKHW
|
||||
CHANNELS_LAST, // Channels-last NHWGC_GKYXC_NHWGK
|
||||
CHANNELS_FIRST // Channels-first NGCHW_GKCYX_NGKHW
|
||||
};
|
||||
|
||||
// Spatial dimensionalities of grouped convolutions.
|
||||
@@ -27,20 +27,20 @@ concept ConvDataType = (T == DataType::FP32) || (T == DataType::FP16) || (T == D
|
||||
// Direction of the convolution operation.
|
||||
enum class ConvDirection
|
||||
{
|
||||
Forward,
|
||||
BackwardData,
|
||||
BackwardWeight
|
||||
FORWARD,
|
||||
BACKWARD_DATA,
|
||||
BACKWARD_WEIGHT
|
||||
};
|
||||
|
||||
// Elementwise operation to fuse to convolution.
|
||||
enum class ElementwiseOperation
|
||||
{
|
||||
Bias,
|
||||
BiasClamp,
|
||||
Bilinear,
|
||||
Clamp,
|
||||
Scale,
|
||||
PassThrough
|
||||
BIAS,
|
||||
BIAS_CLAMP,
|
||||
BILINEAR,
|
||||
CLAMP,
|
||||
SCALE,
|
||||
PASS_THROUGH
|
||||
};
|
||||
|
||||
// Operational signature of a convolution.
|
||||
|
||||
@@ -12,7 +12,7 @@ using P = ckb::BlockGemmPipelineVersion;
|
||||
struct ConvSignature
|
||||
{
|
||||
int spatial_dim = 2;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::Forward;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::FORWARD;
|
||||
ckb::GroupConvLayout layout = ckb::GroupConvLayout::CHANNELS_LAST;
|
||||
ckb::DataType data_type = ckb::DataType::FP16;
|
||||
};
|
||||
|
||||
@@ -12,7 +12,7 @@ using P = ckb::BlockGemmPipelineVersion;
|
||||
struct ConvSignature
|
||||
{
|
||||
int spatial_dim = 2;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::Forward;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::FORWARD;
|
||||
ckb::GroupConvLayout layout = ckb::GroupConvLayout::CHANNELS_LAST;
|
||||
ckb::DataType data_type = ckb::DataType::FP16;
|
||||
};
|
||||
|
||||
@@ -12,7 +12,7 @@ using P = ckb::BlockGemmPipelineVersion;
|
||||
struct ConvSignature
|
||||
{
|
||||
int spatial_dim = 3;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::Forward;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::FORWARD;
|
||||
ckb::GroupConvLayout layout = ckb::GroupConvLayout::CHANNELS_LAST;
|
||||
ckb::DataType data_type = ckb::DataType::BF16;
|
||||
};
|
||||
|
||||
@@ -18,7 +18,7 @@ using P = ckb::BlockGemmPipelineVersion;
|
||||
struct ConvSignature
|
||||
{
|
||||
int spatial_dim = 2;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::Forward;
|
||||
ckb::ConvDirection direction = ckb::ConvDirection::FORWARD;
|
||||
ckb::GroupConvLayout layout = ckb::GroupConvLayout::CHANNELS_LAST;
|
||||
ckb::DataType data_type = ckb::DataType::FP16;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user