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

fixed splitk

Browse Source
This commit is contained in:
Jing Zhang
2024-10-21 12:42:15 -07:00
parent 65cfb2a15c
commit 82bb8dde6e
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
View File

@@ -59,7 +59,6 @@ __device__ inline half2_t pki4_to_half2(pk_i4_t q)
int x_h = (x_u8 & 0xf0) << 12;
const int EX = 0x64006400;
const int SUB = 0xE408E408; //-8
int lo = (x_l | x_h) | EX;

4
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3.hpp
View File

@@ -601,7 +601,7 @@ struct GridwiseGemm_xdl_cshuffle_v3
{
if constexpr(is_same_v<tensor_layout::gemm::RowMajor, ALayout>)
{
a_k_split_offset = blockIdx.z * karg.KRead;
a_k_split_offset = blockIdx.z * karg.KRead / APackedSize;
}
else if constexpr(is_same_v<tensor_layout::gemm::ColumnMajor, ALayout>)
{
@@ -614,7 +614,7 @@ struct GridwiseGemm_xdl_cshuffle_v3
}
else if constexpr(is_same_v<tensor_layout::gemm::ColumnMajor, BLayout>)
{
b_k_split_offset = blockIdx.z * karg.KRead;
b_k_split_offset = blockIdx.z * karg.KRead / BPackedSize;
}
if(blockIdx.z < static_cast<uint32_t>(karg.KBatch - 1))