diff --git a/projects/hipblaslt/tensilelite/tests/MXDataGen_test.cpp b/projects/hipblaslt/tensilelite/tests/MXDataGen_test.cpp index 0fed9f0493c..d66b44fcbc0 100644 --- a/projects/hipblaslt/tensilelite/tests/MXDataGen_test.cpp +++ b/projects/hipblaslt/tensilelite/tests/MXDataGen_test.cpp @@ -65,6 +65,7 @@ TEST_P(MXDataGenFP4Test, ZeroFrequencyWithinBounds) std::vector emptyTile; generateMXInput((hipDataType)HIP_R_4F_E2M1, + HIP_R_8F_UE8M0, dataBuffer.data(), scaleBuffer.data(), rows, @@ -132,12 +133,14 @@ TEST_P(MXGeneratorDeterminismTest, GeneratorOutputIsDeterministic) std::vector emptySwizzle, emptyTile; generateMXInput((hipDataType)HIP_R_4F_E2M1, + HIP_R_8F_UE8M0, data1.data(), scale1.data(), rows, cols, rows, isTranspose, emptySwizzle, emptyTile, mxBlock, 1, isMatrixA, "Bounded", -1.f, 1.f); generateMXInput((hipDataType)HIP_R_4F_E2M1, + HIP_R_8F_UE8M0, data2.data(), scale2.data(), rows, cols, rows, isTranspose, emptySwizzle, emptyTile, @@ -201,6 +204,7 @@ TEST_P(MXPreSwizzleTest, ScaleIsPermutationOfUnswizzled) // Generate without preSwizzle generateMXInput((hipDataType)HIP_R_4F_E2M1, + HIP_R_8F_UE8M0, dataNoShuf.data(), scaleNoShuf.data(), rows, cols, rows, @@ -211,6 +215,7 @@ TEST_P(MXPreSwizzleTest, ScaleIsPermutationOfUnswizzled) // Generate with preSwizzle generateMXInput((hipDataType)HIP_R_4F_E2M1, + HIP_R_8F_UE8M0, dataShuf.data(), scaleShuf.data(), rows, cols, rows, diff --git a/projects/hipblaslt/tensilelite/tests/MXScalePadding_test.cpp b/projects/hipblaslt/tensilelite/tests/MXScalePadding_test.cpp index 429f6fbbb3f..ada326a5c5a 100644 --- a/projects/hipblaslt/tensilelite/tests/MXScalePadding_test.cpp +++ b/projects/hipblaslt/tensilelite/tests/MXScalePadding_test.cpp @@ -45,8 +45,8 @@ static ContractionProblemGemm makeMXProblem(size_t M, M, M * N, 0.0); - problem.setMXScaleA(mxBlock); - problem.setMXScaleB(mxBlock); + problem.setMXScaleA(rocisa::DataType::E8, mxBlock); + problem.setMXScaleB(rocisa::DataType::E8, mxBlock); return problem; }