diff --git a/kernels/portable/cpu/op_full.cpp b/kernels/portable/cpu/op_full.cpp index 74b9657204f..30ca7b825fb 100644 --- a/kernels/portable/cpu/op_full.cpp +++ b/kernels/portable/cpu/op_full.cpp @@ -40,7 +40,7 @@ Tensor& full_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_test.cpp b/kernels/test/op_full_test.cpp index 09885ddd991..318d57e3467 100644 --- a/kernels/test/op_full_test.cpp +++ b/kernels/test/op_full_test.cpp @@ -70,7 +70,7 @@ class OpFullOutTest : public OperatorTest { test_ones_out({2, 3, 4}); \ } -ET_FORALL_REALH_TYPES(GENERATE_TEST) +ET_FORALL_REALHBF16_TYPES(GENERATE_TEST) TEST_F(OpFullOutTest, ValueOverflow) { if (torch::executor::testing::SupportedFeatures::get()->is_aten) {