From 69934f24f5812270ae3175328703cfe34890cb3f Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Tue, 21 Jan 2025 15:50:59 -0800 Subject: [PATCH 1/2] Update [ghstack-poisoned] --- kernels/portable/cpu/op_full_like.cpp | 2 +- kernels/test/op_full_like_test.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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..65c3affbf45 100644 --- a/kernels/test/op_full_like_test.cpp +++ b/kernels/test/op_full_like_test.cpp @@ -87,7 +87,7 @@ void OpFullLikeTest::test_full_like_out() { TEST_F(OpFullLikeTest, AllRealOutputPasses) { #define TEST_ENTRY(ctype, dtype) test_full_like_out(); - ET_FORALL_REAL_TYPES_AND(Bool, TEST_ENTRY); + ET_FORALL_REALHBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY } From ad860de60027f857f25a9e9100ef66b7ecacc4b7 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Wed, 22 Jan 2025 09:51:32 -0800 Subject: [PATCH 2/2] Update [ghstack-poisoned] --- kernels/test/op_full_like_test.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kernels/test/op_full_like_test.cpp b/kernels/test/op_full_like_test.cpp index 65c3affbf45..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_REALHBF16_TYPES(TEST_ENTRY); + ET_FORALL_REALHBBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY }