ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-15 11:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update GetTypeString for grouped bwd wei explicit gemm

Browse Source
This commit is contained in:
Bartlomiej Kocot
2025-07-29 14:47:36 +00:00
parent d0364641ed
commit fe303e69b6
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_explicit_xdl.hpp
View File

@@ -6,6 +6,7 @@
#include <iostream>
#include <numeric>
#include <sstream>
#include <string>
#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
{