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
|
||||
#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
|
||||
for (int lh = 0; lh < Z_REMAINDER_SIZE; ++lh) {
|
||||
// read
|
||||
|
@ -1913,7 +1913,8 @@ INSTANTIATE_TEST_SUITE_P(, permute_bfzyx_to_bfyxz,
|
||||
::testing::ValuesIn(std::vector<TiledPermuteParam> {
|
||||
{{1, 3, 85, 20, 20}, 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) {
|
||||
|
Loading…
Reference in New Issue
Block a user