fix a bug in permute_bfzyx_to_bfyxz (#16599)
This commit is contained in:
parent
3849d5aa02
commit
e169c7cd38
@ -102,7 +102,7 @@ KERNEL (permute_bfzyx_to_bfyxz)(
|
|||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
#if defined(X_REMAINDER_ITEM) && defined(Z_REMAINDER_ITEM)
|
#if defined(X_REMAINDER_ITEM) && defined(Z_REMAINDER_ITEM)
|
||||||
else if (f == Z_REMAINDER_ITEM && x == X_REMAINDER_ITEM) {
|
else if (z == Z_REMAINDER_ITEM && x == X_REMAINDER_ITEM) {
|
||||||
// point by point
|
// point by point
|
||||||
for (int lh = 0; lh < Z_REMAINDER_SIZE; ++lh) {
|
for (int lh = 0; lh < Z_REMAINDER_SIZE; ++lh) {
|
||||||
// read
|
// read
|
||||||
|
@ -1913,7 +1913,8 @@ INSTANTIATE_TEST_SUITE_P(, permute_bfzyx_to_bfyxz,
|
|||||||
::testing::ValuesIn(std::vector<TiledPermuteParam> {
|
::testing::ValuesIn(std::vector<TiledPermuteParam> {
|
||||||
{{1, 3, 85, 20, 20}, format::bfzyx},
|
{{1, 3, 85, 20, 20}, format::bfzyx},
|
||||||
{{1, 3, 85, 40, 40}, format::bfzyx},
|
{{1, 3, 85, 40, 40}, format::bfzyx},
|
||||||
{{1, 3, 85, 80, 80}, format::bfzyx}
|
{{1, 3, 85, 80, 80}, format::bfzyx},
|
||||||
|
{{1, 192, 2, 64, 33}, format::bfzyx}
|
||||||
}));
|
}));
|
||||||
|
|
||||||
TEST_P(permute_bfzyx_to_bfyxz, combined) {
|
TEST_P(permute_bfzyx_to_bfyxz, combined) {
|
||||||
|
Loading…
Reference in New Issue
Block a user