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

Grouped conv fwd v3 fix for SplitN an G > 1 (#2038)

Browse Source 
* Grouped conv fwd v3 fix for SplitN an G > 1

* Remove int8 large test

* Retore int8 test

[ROCm/composable_kernel commit: ec742908bd]
This commit is contained in:
Bartłomiej Kocot
2025-04-01 22:19:35 +02:00
committed by GitHub
parent 0ac4c37028
commit b417137276
2 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/grouped_convnd_fwd/test_grouped_convnd_fwd_large_cases_xdl.cpp
View File

@@ -1,5 +1,5 @@
// SPDX-License-Identifier: MIT
// Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
#include <cstdlib>
#include <iostream>
@@ -83,6 +83,9 @@ TYPED_TEST(TestGroupedConvndFwd2d, Test2D)
// When image is larger than 2GB
this->conv_params.push_back(
{2, 2, 2, 128, 128, {3, 3}, {4096, 2048}, {300, 300}, {3, 3}, {1, 1}, {1, 1}});
// Split N and G > 1
this->conv_params.push_back(
{2, 4, 112, 8, 8, {3, 3}, {469, 724}, {2, 2}, {2, 2}, {1, 1}, {1, 1}});
this->template Run<2>();
}