diff --git a/test/mx_mfma_op/mx_mfma_op.hpp b/test/mx_mfma_op/mx_mfma_op.hpp index b2e615b9d8..47b4419379 100644 --- a/test/mx_mfma_op/mx_mfma_op.hpp +++ b/test/mx_mfma_op/mx_mfma_op.hpp @@ -789,12 +789,12 @@ struct store_C_col_major CScalarFragT chunks[vectorSize(CFragT{}) / VW]; } fragC{cFrag}; // Initialize with input fragment - *(reinterpret_cast(output + startOffset)) = fragC.chunks[0]; - *(reinterpret_cast(output + startOffset + kMajorOffset)) = fragC.chunks[1]; - *(reinterpret_cast(output + startOffset + 2 * kMajorOffset)) = - fragC.chunks[2]; - *(reinterpret_cast(output + startOffset + 3 * kMajorOffset)) = - fragC.chunks[3]; + CScalarFragT* fragPtr; + for(uint32_t idx = 0; idx < vectorSize(CFragT{}) / VW; ++idx) + { + fragPtr = reinterpret_cast(output + startOffset + idx * kMajorOffset); + *fragPtr = fragC.chunks[idx]; + } } };