Support BFloat16 in full_like

swolchok · swolchok · commit fa3e6940a80c · 2025-01-21T15:51:03.000-08:00
Partial fix for #7748. ghstack-source-id: 0c618df ghstack-comment-id: 2605975252 Pull Request resolved: #7822
diff --git 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<CTYPE_OUT>(val);
       auto data_out = out.mutable_data_ptr<CTYPE_OUT>();
       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
@@ -87,7 +87,7 @@ void OpFullLikeTest::test_full_like_out<ScalarType::Bool>() {
 
 TEST_F(OpFullLikeTest, AllRealOutputPasses) {
 #define TEST_ENTRY(ctype, dtype) test_full_like_out<ScalarType::dtype>();
-  ET_FORALL_REAL_TYPES_AND(Bool, TEST_ENTRY);
+  ET_FORALL_REALHBF16_TYPES(TEST_ENTRY);
 #undef TEST_ENTRY
 }
 

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ void OpFullLikeTest::test_full_like_out<ScalarType::Bool>() {`
`87`	`87`
`88`	`88`	`TEST_F(OpFullLikeTest, AllRealOutputPasses) {`
`89`	`89`	`#define TEST_ENTRY(ctype, dtype) test_full_like_out<ScalarType::dtype>();`
`90`		`- ET_FORALL_REAL_TYPES_AND(Bool, TEST_ENTRY);`
	`90`	`+ ET_FORALL_REALHBF16_TYPES(TEST_ENTRY);`
`91`	`91`	`#undef TEST_ENTRY`
`92`	`92`	`}`
`93`	`93`