diff --git a/tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc b/tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc index 25caa741..6217c813 100644 --- a/tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc +++ b/tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc @@ -6897,5 +6897,20 @@ TEST_F(UtestFormatTransferFracZHwcn, fp16_1c_1n_pad_cn) { EXPECT_EQ((reinterpret_cast(result.data.get()))[i], ret[i]); } } +TEST_F(UtestFormatTransferFracZHwcn, fracz_to_hwcn_fp16_success_with_groups) { + uint16_t data_4d[2 * 2 * 2 * 2] = {1}; + uint16_t data[1 * 1 * 1 * 2 *2 * 16 *16] = {1024}; + int64_t groups = 1; + TransArgs args{ + reinterpret_cast(data_4d), FORMAT_FRACTAL_Z, FORMAT_HWCN, {1, 1, 1, 2, 2, 16, 6}, {2, 2, 2, 2}, DT_FLOAT16 ,groups}; + TransResult result; + + FormatTransferFracZHwcn transfer; + EXPECT_EQ(transfer.TransFormat(args, result), SUCCESS); + EXPECT_EQ(result.length, sizeof(data)); + for (int i = 0; i < sizeof(data) / sizeof(data[0]); ++i) { + EXPECT_EQ((reinterpret_cast(result.data.get()))[i], data[i]); + } +} } // namespace formats } // namespace ge