From 7951ca51d630f61acebcd7f945ab5dac2d51f9c2 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Tue, 21 Jan 2025 15:46:57 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- kernels/portable/cpu/op_full.cpp | 2 +- kernels/test/op_full_test.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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) {