diff --git a/kernels/portable/cpu/op_full_like.cpp b/kernels/portable/cpu/op_full_like.cpp index 682a834b9cd..6cb64cebbde 100644 --- a/kernels/portable/cpu/op_full_like.cpp +++ b/kernels/portable/cpu/op_full_like.cpp @@ -56,7 +56,7 @@ Tensor& full_like_out( CTYPE_VAL val; utils::extract_scalar(fill_value, &val); - ET_SWITCH_REALHB_TYPES(out_type, ctx, name, CTYPE_OUT, [&] { + ET_SWITCH_REALHBBF16_TYPES(out_type, ctx, name, CTYPE_OUT, [&] { CTYPE_OUT val_casted = static_cast(val); auto data_out = out.mutable_data_ptr(); for (size_t i = 0; i < out.numel(); ++i) { diff --git a/kernels/test/op_full_like_test.cpp b/kernels/test/op_full_like_test.cpp index 1a9c3c0a7af..6a892c81602 100644 --- a/kernels/test/op_full_like_test.cpp +++ b/kernels/test/op_full_like_test.cpp @@ -85,9 +85,9 @@ void OpFullLikeTest::test_full_like_out() { EXPECT_TENSOR_EQ(out, tf.zeros(sizes)); } -TEST_F(OpFullLikeTest, AllRealOutputPasses) { +TEST_F(OpFullLikeTest, AllDtypeOutputPasses) { #define TEST_ENTRY(ctype, dtype) test_full_like_out(); - ET_FORALL_REAL_TYPES_AND(Bool, TEST_ENTRY); + ET_FORALL_REALHBBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY }