From 9c34128c3ec36cb9e5fb010a99e2a80b41ac38b8 Mon Sep 17 00:00:00 2001 From: Bartlomiej Kocot Date: Tue, 29 Jul 2025 14:47:36 +0000 Subject: [PATCH] Update GetTypeString for grouped bwd wei explicit gemm --- ...evice_grouped_conv_bwd_weight_explicit_xdl.hpp | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_explicit_xdl.hpp b/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_explicit_xdl.hpp index a819b91b05..5f116d0029 100644 --- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_explicit_xdl.hpp +++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_explicit_xdl.hpp @@ -6,6 +6,7 @@ #include #include #include +#include #include "ck/utility/common_header.hpp" @@ -443,14 +444,14 @@ struct DeviceGroupedConvBwdWeight_Explicit_Xdl std::string GetTypeString() const override { - auto str = std::stringstream(); + std::string gemm_type_string = DeviceGemmV3Op{}.GetTypeString(); + gemm_type_string = std::regex_replace(gemm_type_string, + std::regex("DeviceBatchedGemmXdlUniversal"), + "DeviceGroupedConvBwdWeight_Explicit_Xdl"); + gemm_type_string = std::regex_replace(gemm_type_string, std::regex(">"), ","); + gemm_type_string.append(">"); - // clang-format off - str << "DeviceGroupedConvBwdWeight_Explicit_Xdl" - << "<" << DeviceGemmV3Op{}.GetTypeString() << ">"; - // clang-format on - - return str.str(); + return gemm_type_string; } size_t GetWorkSpaceSize(const BaseArgument* p_arg) const override {