Commit 2efa20ad authored by Bartlomiej Wroblewski's avatar Bartlomiej Wroblewski
Browse files

Review: Rename mfma_input_type to dpp_input_type

parent 4bb61e35
...@@ -279,13 +279,13 @@ struct BlockwiseGemmDpp_k0mk1_k0nk1_m0n0m1n1m2n2 ...@@ -279,13 +279,13 @@ struct BlockwiseGemmDpp_k0mk1_k0nk1_m0n0m1n1m2n2
[Number<b_thread_desc_.CalculateOffset(make_tuple(0, 0, 0, k + i))>{}]; [Number<b_thread_desc_.CalculateOffset(make_tuple(0, 0, 0, k + i))>{}];
}); });
using mfma_input_type = typename vector_type<FloatAB, dpp_gemm.K1PerDpp>::type; using dpp_input_type = typename vector_type<FloatAB, dpp_gemm.K1PerDpp>::type;
constexpr index_t c_offset = constexpr index_t c_offset =
c_thread_desc_.CalculateOffset(make_tuple(m0, n0, 0)); c_thread_desc_.CalculateOffset(make_tuple(m0, n0, 0));
dpp_gemm.template Run(a_thread_vec.template AsType<mfma_input_type>(), dpp_gemm.template Run(a_thread_vec.template AsType<dpp_input_type>(),
b_thread_vec.template AsType<mfma_input_type>(), b_thread_vec.template AsType<dpp_input_type>(),
c_thread_buf.GetVectorTypeReference(Number<c_offset>{})); c_thread_buf.GetVectorTypeReference(Number<c_offset>{}));
}); });
}); });
......
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