pull/1281/head
zk 4 years ago
parent 459c9a4ab2
commit aef76524d4

@ -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

Loading…
Cancel
Save