ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 14:59:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.

Browse Source 
Signed-off-by: Michal Kulikowski <Michal.Kulikowski@amd.com>
This commit is contained in:
Michal Kulikowski
2025-10-16 13:01:24 +02:00
committed by Michał Kulikowski
parent 0d3860dfdb
commit b9789a0742
33 changed files with 195 additions and 199 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/31_batched_gemm_gemm/run_batched_gemm_gemm_example.inc
View File

@@ -1,8 +1,10 @@
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
using Bypass = ck::tensor_layout::BypassLayoutVerification;
bool run_batched_gemm_gemm_example(int argc, char* argv[])
{
bool do_verification = true;
@@ -111,12 +113,12 @@ bool run_batched_gemm_gemm_example(int argc, char* argv[])
if(std::is_same<decltype(layout), Row>::value)
{
return HostTensorDescriptor(
{batch_count, row, col}, {batch_stride, stride, 1_uz}, layout);
{batch_count, row, col}, {batch_stride, stride, 1_uz}, Bypass{});
}
else
{
return HostTensorDescriptor(
{batch_count, row, col}, {batch_stride, 1_uz, stride}, layout);
{batch_count, row, col}, {batch_stride, 1_uz, stride}, Bypass{});
}
};