Half support for index op

Reviewed By: cccclai

Differential Revision: D56543186

fbshipit-source-id: 4fed6b9b3ede3cdcb67a9a52150e3f22cc02b180

Author: manuelcandales

kernels/portable/cpu/op_index.cpp

@@ -40,12 +40,11 @@ Tensor& index_Tensor_out(
   if (block_count == 0) {
     ET_KERNEL_CHECK(
         ctx, resize_tensor(out, in.sizes()) == Error::Ok, InvalidArgument, out);
-    ET_SWITCH_REAL_TYPES_AND(
-        Bool, in_type, ctx, "index.Tensor_out", CTYPE, [&]() {
-          const CTYPE* const in_data = in.const_data_ptr<CTYPE>();
-          CTYPE* const out_data = out.mutable_data_ptr<CTYPE>();
-          memcpy(out_data, in_data, in.nbytes());
-        });
+    ET_SWITCH_REALHB_TYPES(in_type, ctx, "index.Tensor_out", CTYPE, [&]() {
+      const CTYPE* const in_data = in.const_data_ptr<CTYPE>();
+      CTYPE* const out_data = out.mutable_data_ptr<CTYPE>();
+      memcpy(out_data, in_data, in.nbytes());
+    });
     return out;
   }
 
@@ -85,20 +84,19 @@ Tensor& index_Tensor_out(
   compute_dim_map(in, indices, dim_map, block_count == 1);
   compute_index_map(in, indices, ix_map);
 
-  ET_SWITCH_REAL_TYPES_AND(
-      Bool, in_type, ctx, "index.Tensor_out", CTYPE, [&]() {
-        const CTYPE* const in_data = in.const_data_ptr<CTYPE>();
-        CTYPE* const out_data = out.mutable_data_ptr<CTYPE>();
-
-        for (auto out_ix = 0; out_ix < out.numel(); out_ix++) {
-          size_t in_ix = 0;
-          bool success = true;
-          std::tie(in_ix, success) =
-              get_in_ix(in, indices, out, out_ix, start, xdim, dim_map, ix_map);
-          ET_KERNEL_CHECK(ctx, success, InvalidArgument, );
-          out_data[out_ix] = in_data[in_ix];
-        }
-      });
+  ET_SWITCH_REALHB_TYPES(in_type, ctx, "index.Tensor_out", CTYPE, [&]() {
+    const CTYPE* const in_data = in.const_data_ptr<CTYPE>();
+    CTYPE* const out_data = out.mutable_data_ptr<CTYPE>();
+
+    for (auto out_ix = 0; out_ix < out.numel(); out_ix++) {
+      size_t in_ix = 0;
+      bool success = true;
+      std::tie(in_ix, success) =
+          get_in_ix(in, indices, out, out_ix, start, xdim, dim_map, ix_map);
+      ET_KERNEL_CHECK(ctx, success, InvalidArgument, );
+      out_data[out_ix] = in_data[in_ix];
+    }
+  });
 
   return out;
 }