fix a bug in permute_bfzyx_to_bfyxz (#16599)

This commit is contained in:
Eddy Kim 2023-03-29 03:19:35 +09:00 committed by GitHub
parent 3849d5aa02
commit e169c7cd38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

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

View File

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