From 864168931263429f44cf562ab3def68f86ca0f06 Mon Sep 17 00:00:00 2001 From: ginchang Date: Tue, 25 Mar 2025 15:54:06 +0800 Subject: [PATCH] [ckProfiler] Add bf16/int8 support on groupConv 1. Support NGCHW groupConv fwd data --- profiler/src/profile_grouped_conv_fwd.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/profiler/src/profile_grouped_conv_fwd.cpp b/profiler/src/profile_grouped_conv_fwd.cpp index 7faf573dbf..9afe3eb32c 100644 --- a/profiler/src/profile_grouped_conv_fwd.cpp +++ b/profiler/src/profile_grouped_conv_fwd.cpp @@ -302,6 +302,14 @@ int profile_grouped_conv_fwd(int argc, char* argv[]) { return profile(I2, NGCHW{}, GKYXC{}, NGKHW{}, F16{}, F16{}, F16{}, F16{}, F16{}); } + else if(data_type == ConvDataType::BF16_BF16_BF16) + { + return profile(I2, NGCHW{}, GKYXC{}, NGKHW{}, BF16{}, BF16{}, BF16{}, BF16{}, BF16{}); + } + else if(data_type == ConvDataType::INT8_INT8_INT8) + { + return profile(I2, NGCHW{}, GKYXC{}, NGKHW{}, INT8{}, INT8{}, INT8{}, INT8{}, INT8{}); + } } else if(num_dim_spatial == 3 && layout == ConvLayout::NHWGC_GKYXC_NHWGK) {