|
|
|
@ -34488,24 +34488,6 @@ TEST_F(UtestFormatTransferHwcnFz, build_transfer_int8) {
|
|
|
|
|
EXPECT_NE(transfer, nullptr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestFormatTransferHwcnFz, build_transfer_int8) {
|
|
|
|
|
int8_t data[4 * 4 * 3 * 1];
|
|
|
|
|
TransArgs args{
|
|
|
|
|
reinterpret_cast<uint8_t *>(data), FORMAT_HWCN, FORMAT_FRACTAL_Z, std::vector<int64_t>({4, 4, 3, 1}),
|
|
|
|
|
std::vector<int64_t>({16, 1, 16, 32}), DT_INT8};
|
|
|
|
|
auto transfer = BuildFormatTransfer(args);
|
|
|
|
|
EXPECT_NE(transfer, nullptr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestFormatTransferHwcnFz, build_transfer_int8) {
|
|
|
|
|
int8_t data[4 * 4 * 3 * 1];
|
|
|
|
|
TransArgs args{
|
|
|
|
|
reinterpret_cast<uint8_t *>(data), FORMAT_HWCN, FORMAT_FRACTAL_Z, std::vector<int64_t>({4, 4, 3, 1}),
|
|
|
|
|
std::vector<int64_t>({16, 1, 16, 32}), DT_INT8};
|
|
|
|
|
auto transfer = BuildFormatTransfer(args);
|
|
|
|
|
EXPECT_NE(transfer, nullptr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestFormatTransferHwcnFz, build_transfer_not_support) {
|
|
|
|
|
float data[50 * 2 * 16 * 16];
|
|
|
|
|
TransArgs args{
|
|
|
|
@ -34514,14 +34496,5 @@ TEST_F(UtestFormatTransferHwcnFz, build_transfer_not_support) {
|
|
|
|
|
auto transfer = BuildFormatTransfer(args);
|
|
|
|
|
EXPECT_EQ(transfer, nullptr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestFormatTransferHwcnFz, build_transfer_int8_with_groups) {
|
|
|
|
|
int8_t data[4 * 4 * 3 * 1];
|
|
|
|
|
TransArgs args{
|
|
|
|
|
reinterpret_cast<uint8_t *>(data), FORMAT_HWCN, FORMAT_FRACTAL_Z, std::vector<int64_t>({4, 4, 3, 1}),
|
|
|
|
|
std::vector<int64_t>({16, 1, 16, 32}), DT_INT8};
|
|
|
|
|
auto transfer = BuildFormatTransfer(args);
|
|
|
|
|
EXPECT_NE(transfer, nullptr);
|
|
|
|
|
}
|
|
|
|
|
} // namespace formats
|
|
|
|
|
} // namespace ge
|
|
|
|
|