From 6ca9d532c18a39c76c7936e09012858daf4c0526 Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Thu, 23 Jan 2025 13:39:10 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- kernels/portable/cpu/op_unbind_copy.cpp | 8 ++++---- kernels/test/op_unbind_copy_test.cpp | 6 +++--- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/kernels/portable/cpu/op_unbind_copy.cpp b/kernels/portable/cpu/op_unbind_copy.cpp index 1f4930bacd0..ba33b87e759 100644 --- a/kernels/portable/cpu/op_unbind_copy.cpp +++ b/kernels/portable/cpu/op_unbind_copy.cpp @@ -54,10 +54,10 @@ void unbind_copy_int_out( ScalarType in_type = input.scalar_type(); ScalarType out_type = out[0].scalar_type(); - ET_SWITCH_REAL_TYPES_AND( - Bool, in_type, ctx, "unbind_copy.int_out", CTYPE_IN, [&]() { - ET_SWITCH_REAL_TYPES_AND( - Bool, out_type, ctx, "unbind_copy.int_out", CTYPE_OUT, [&]() { + ET_SWITCH_REALHBF16_TYPES( + in_type, ctx, "unbind_copy.int_out", CTYPE_IN, [&]() { + ET_SWITCH_REALHBF16_TYPES( + out_type, ctx, "unbind_copy.int_out", CTYPE_OUT, [&]() { const CTYPE_IN* const input_data = input.const_data_ptr(); for (size_t i = 0, e = out.size(); i < e; ++i) { diff --git a/kernels/test/op_unbind_copy_test.cpp b/kernels/test/op_unbind_copy_test.cpp index 4901b0f69e3..26aa0a7f9c6 100644 --- a/kernels/test/op_unbind_copy_test.cpp +++ b/kernels/test/op_unbind_copy_test.cpp @@ -208,19 +208,19 @@ class OpUnbindCopyIntOutTest : public OperatorTest { */ TEST_F(OpUnbindCopyIntOutTest, Unbind1x2x3OnDim0AllRealDtypes) { #define TEST_ENTRY(ctype, dtype) test_unbind_dim0(); - ET_FORALL_REAL_TYPES(TEST_ENTRY); + ET_FORALL_REALHBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY } TEST_F(OpUnbindCopyIntOutTest, Unbind1x2x3OnDim1AllRealDTypes) { #define TEST_ENTRY(ctype, dtype) test_unbind_dim1(); - ET_FORALL_REAL_TYPES(TEST_ENTRY); + ET_FORALL_REALHBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY } TEST_F(OpUnbindCopyIntOutTest, Unbind1x2x3OnDim2AllRealDTypes) { #define TEST_ENTRY(ctype, dtype) test_unbind_dim2(); - ET_FORALL_REAL_TYPES(TEST_ENTRY); + ET_FORALL_REALHBF16_TYPES(TEST_ENTRY); #undef TEST_ENTRY }