revise

intel · Jul 10, 2024 · f70f9ec · f70f9ec
1 parent 87452c6
commit f70f9ec
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 18 deletions.
diff --git a/src/ATen/native/xpu/TensorAdvancedIndexing.cpp b/src/ATen/native/xpu/TensorAdvancedIndexing.cpp
@@ -342,10 +342,6 @@ Tensor& XPUNativeFunctions::index_fill_(
 
   // Handle the case when `self` is 0-dim
   Tensor self_nonzero_dim = (self.dim() == 0) ? self.unsqueeze(-1) : self;
-
-  dim = maybe_wrap_dim(dim, self.dim());
-  TORCH_CHECK(index.dim() <= 1, "Index has to be a vector/scalar");
-  index_func_meta_impl(self, self, dim, index, self, "index_fill");
   native::xpu::index_fill_kernel(self, dim, index, source, self);
   return self;
 }

diff --git a/src/ATen/native/xpu/sycl/Indexing.cpp b/src/ATen/native/xpu/sycl/Indexing.cpp
@@ -420,18 +420,6 @@ struct IndexFillScalarFunctor {
     dst[dst_off] = alpha;
   }
 };
-template <>
-struct IndexFillScalarFunctor<bool> {
-  void operator()(
-      bool* dst,
-      bool* src,
-      int64_t dst_off,
-      int64_t src_off,
-      int64_t idx,
-      bool alpha) const {
-    dst[dst_off] = alpha;
-  }
-};
 
 void index_fill_kernel(
     const Tensor& self,

diff --git a/yaml/xpu_functions.yaml b/yaml/xpu_functions.yaml
@@ -87,9 +87,7 @@ supported:
   - index_add_
   - index_add
   - index_fill_.int_Tensor
-  - index_fill_.int_Scalar
   - index_fill.int_Tensor
-  - index_fill.int_Scalar
   - index_select
   - index_select.out
   - gcd