fix a bug in permute_bfzyx_to_bfyxz (#16599)

This commit is contained in:
Eddy Kim authored and GitHub committed 2023-03-28 18:19:35 +00:00
1 parent 3849d5aa02
commit e169c7cd38
2 files changed
+3 -2

No files matched your search

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