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 be94da1e50..102673810a 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" @@ -491,14 +492,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 {