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

Add grouped conv fwd 3d GKCYX instances for f32, f16, bf16 (#2069)

Browse Source 
* Part1

* Add grouped conv fwd 3d GKCYX instances for f32, f16, bf16

* Add missing coma

* Add missing cpp instance files

* Fix 3d layout

* Add missing closing bracket

* Add missing comp x2 and part2 instances

* Fix typo in instance name

* fix

* Fix

---------

Co-authored-by: Bartlomiej Kocot <barkocot@amd.com>

[ROCm/composable_kernel commit: 7c32652e03]
This commit is contained in:
aledudek
2025-04-16 11:00:55 +02:00
committed by GitHub
parent 4ec293cb4b
commit c26c2b1fdc
28 changed files with 1377 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/grouped_convnd_fwd/test_grouped_convnd_fwd.cpp
View File

@@ -77,7 +77,9 @@ using KernelTypes3d = ::testing::Types<std::tuple<float, GNDHWC, GKZYXC, GNDHWK>
std::tuple<float, NDHWGC, GKZYXC, NDHWGK>,
std::tuple<ck::half_t, NDHWGC, GKZYXC, NDHWGK>,
std::tuple<ck::bhalf_t, NDHWGC, GKZYXC, NDHWGK>,
std::tuple<int8_t, NDHWGC, GKZYXC, NDHWGK>>;
std::tuple<float, NGCDHW, GKCZYX, NGKDHW>,
std::tuple<ck::half_t, NGCDHW, GKCZYX, NGKDHW>,
std::tuple<ck::bhalf_t, NGCDHW, GKCZYX, NGKDHW>>;
template <typename Tuple>
class TestGroupedConvndFwd1d : public TestGroupedConvndFwd<Tuple>