Commit 05672519 authored by Bartlomiej Wroblewski's avatar Bartlomiej Wroblewski
Browse files

Review: Align struct field names; Use Number instead of index_t

parent 25907115
......@@ -273,9 +273,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
block_2_ctile_map_{},
M01_{M01},
N01_{N01},
M_raw{M},
N_raw{N},
K_raw{K},
M_raw_{M},
N_raw_{N},
K_raw_{K},
a_element_op_{a_element_op},
b_element_op_{b_element_op},
c_element_op_{c_element_op}
......@@ -317,9 +317,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
index_t M01_;
index_t N01_;
index_t M_raw;
index_t N_raw;
index_t K_raw;
index_t M_raw_;
index_t N_raw_;
index_t K_raw_;
// TODO: unused since gridwise_gemm_dl_v1r3 does NOT support prologue for the time being.
AElementwiseOperation a_element_op_;
......@@ -497,9 +497,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
if constexpr(is_same<tensor_layout::gemm::RowMajor, ALayout>::value)
{
constexpr auto A_K_vec_length =
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(0) *
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(3);
if(arg.K_raw % A_K_vec_length != 0)
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(I0) *
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(I3);
if(arg.K_raw_ % A_K_vec_length != 0)
{
return false;
}
......@@ -507,9 +507,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
else
{
constexpr auto A_M_vec_lenght =
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(1) *
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(2);
if(arg.M_raw % A_M_vec_lenght != 0)
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(I1) *
ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1::At(I2);
if(arg.M_raw_ % A_M_vec_lenght != 0)
{
return false;
}
......@@ -518,9 +518,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
if constexpr(is_same<tensor_layout::gemm::RowMajor, BLayout>::value)
{
constexpr auto B_N_vec_lenght =
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(1) *
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(2);
if(arg.N_raw % B_N_vec_lenght != 0)
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(I1) *
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(I2);
if(arg.N_raw_ % B_N_vec_lenght != 0)
{
return false;
}
......@@ -528,9 +528,9 @@ struct DeviceGemmDl : public DeviceGemm<ALayout,
else
{
constexpr auto B_K_vec_length =
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(0) *
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(3);
if(arg.K_raw % B_K_vec_length != 0)
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(I0) *
BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1::At(I3);
if(arg.K_raw_ % B_K_vec_length != 0)
{
return false;
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment