Skip to content

Commit e81f371

Browse files
swolchokZonglin Peng
authored andcommitted
Support BFloat16 in full_like (pytorch#7822)
Partial fix for pytorch#7748.
1 parent 0ab6e8a commit e81f371

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

kernels/portable/cpu/op_full_like.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ Tensor& full_like_out(
5656
CTYPE_VAL val;
5757
utils::extract_scalar(fill_value, &val);
5858

59-
ET_SWITCH_REALHB_TYPES(out_type, ctx, name, CTYPE_OUT, [&] {
59+
ET_SWITCH_REALHBBF16_TYPES(out_type, ctx, name, CTYPE_OUT, [&] {
6060
CTYPE_OUT val_casted = static_cast<CTYPE_OUT>(val);
6161
auto data_out = out.mutable_data_ptr<CTYPE_OUT>();
6262
for (size_t i = 0; i < out.numel(); ++i) {

kernels/test/op_full_like_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,9 @@ void OpFullLikeTest::test_full_like_out<ScalarType::Bool>() {
8585
EXPECT_TENSOR_EQ(out, tf.zeros(sizes));
8686
}
8787

88-
TEST_F(OpFullLikeTest, AllRealOutputPasses) {
88+
TEST_F(OpFullLikeTest, AllDtypeOutputPasses) {
8989
#define TEST_ENTRY(ctype, dtype) test_full_like_out<ScalarType::dtype>();
90-
ET_FORALL_REAL_TYPES_AND(Bool, TEST_ENTRY);
90+
ET_FORALL_REALHBBF16_TYPES(TEST_ENTRY);
9191
#undef TEST_ENTRY
9292
}
9393

0 commit comments

Comments
 (0)