Add aten::is_set_to (#516)

Signed-off-by: Feng Yuan <feng1.yuan@intel.com>
intel · Jul 8, 2024 · 441a56c · 441a56c
1 parent 0fe12f3
commit 441a56c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/ATen/native/xpu/TensorProperties.cpp b/src/ATen/native/xpu/TensorProperties.cpp
@@ -0,0 +1,16 @@
+#include <ATen/xpu/XPUNativeFunctions.h>
+
+#ifndef AT_PER_OPERATOR_HEADERS
+#include <ATen/Functions.h>
+#include <ATen/NativeFunctions.h>
+#else
+#include <ATen/ops/is_set_to_native.h>
+#endif
+
+namespace at {
+
+bool XPUNativeFunctions::is_set_to(const Tensor& self, const Tensor& src) {
+  return at::native::is_set_to(self, src);
+}
+
+} // namespace at
diff --git a/yaml/xpu_functions.yaml b/yaml/xpu_functions.yaml
@@ -455,6 +455,7 @@ supported:
   - _cdist_forward
   - _pin_memory
   - is_pinned
+  - is_set_to
   - bucketize.Tensor
   - bucketize.Tensor_out
   - bucketize.Scalar