ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-21 21:39:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '3348f01e6fc65a7afcea3ea4167cc70e902e854a' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-10-15 15:12:32 +00:00
parent d19ba3362d
commit bcc168f1f9
7 changed files with 126 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
example/ck_tile/20_grouped_convolution/gemm_configs.hpp
View File

@@ -226,20 +226,20 @@ struct ConvTypeConfig;
template <>
struct ConvTypeConfig<ck_tile::half_t>
{
using InDataType = ck_tile::half_t;
using WeiDataType = ck_tile::half_t;
using InDataType = ck_tile::half_t;
using WeiDataType = ck_tile::half_t;
using AccDataType = float;
using OutDataType = ck_tile::half_t;
using OutDataType = ck_tile::half_t;
// ToDo: Add more bias config to support different categories of GEMM.
};
template <>
struct ConvTypeConfig<ck_tile::bf16_t, ck_tile::bf16_t, ck_tile::bf16_t>
{
using InDataType = ck_tile::bf16_t;
using WeiDataType = ck_tile::bf16_t;
using InDataType = ck_tile::bf16_t;
using WeiDataType = ck_tile::bf16_t;
using AccDataType = float;
using OutDataType = ck_tile::bf16_t;
using OutDataType = ck_tile::bf16_t;
};
template <typename T>