diff --git a/kernels/test/op_pixel_shuffle_test.cpp b/kernels/test/op_pixel_shuffle_test.cpp index a255f94b2ea..9a111a106f1 100644 --- a/kernels/test/op_pixel_shuffle_test.cpp +++ b/kernels/test/op_pixel_shuffle_test.cpp @@ -66,7 +66,7 @@ TEST_F(OpPixelShuffleOutTest, AllRealDtypesSupported) { #define ENUMERATE_TEST_ENTRY(ctype, dtype) \ test_pixel_shuffle(); - ET_FORALL_REAL_TYPES(ENUMERATE_TEST_ENTRY) + ET_FORALL_REALHBBF16_TYPES(ENUMERATE_TEST_ENTRY) #undef ENUMERATE_TEST_ENTRY } diff --git a/kernels/test/op_pixel_unshuffle_test.cpp b/kernels/test/op_pixel_unshuffle_test.cpp index 838aa4c6946..94fc11fa555 100644 --- a/kernels/test/op_pixel_unshuffle_test.cpp +++ b/kernels/test/op_pixel_unshuffle_test.cpp @@ -65,7 +65,7 @@ TEST_F(OpPixelUnshuffleOutTest, AllRealDtypesSupported) { #define ENUMERATE_TEST_ENTRY(ctype, dtype) \ test_pixel_unshuffle(); - ET_FORALL_REAL_TYPES(ENUMERATE_TEST_ENTRY) + ET_FORALL_REALHBBF16_TYPES(ENUMERATE_TEST_ENTRY) #undef ENUMERATE_TEST_ENTRY }