[rocm-libraries] ROCm/rocm-libraries#5387 (commit 0c259bd)

[CK][CK Tile] Grouped Convolution Backward Weight set of
 fixes (#5387)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## Motivation

Grouped Convolution Backward Weight split k fixes for CK tile kernels

## Technical Details

- get k batch from kargs to get deduced k batch
- multiply zeroing size by data type size
- disable v6 (producing a incorrect results)

## Test Plan

test_grouped_convnd_bwd_weight_tile

## Test Result

Pass

## Submission Checklist

- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
This commit is contained in:
Bartłomiej Kocot
2026-03-13 16:19:50 +00:00
committed by assistant-librarian[bot]
parent 574c1c121a
commit b8108662da
6 changed files with 28 additions and 11 deletions

View File

@@ -126,7 +126,7 @@ struct GroupedConvolutionBackwardWeightInvoker
}
auto preprocess = [&]() {
if(args.k_batch > 1)
if(kargs.k_batch > 1)
{
ck_tile::hip_check_error(hipMemsetAsync(
kargs.wei_ptr, 0, args.template GetWeightByte<WeiDataType>(), s.stream_id_));

View File

@@ -180,7 +180,7 @@ struct GroupedConvolutionBackwardWeightTwoStageInvoker
}
auto preprocess = [&]() {
if(args.k_batch > 1)
if(kargs.k_batch > 1)
ck_tile::hip_check_error(
hipMemsetAsync(ws_args.wei_ptr,
0,