Skip to content

Commit

Permalink
Add aten::is_set_to
Browse files Browse the repository at this point in the history
Signed-off-by: Feng Yuan <[email protected]>
  • Loading branch information
fengyuan14 committed Jul 1, 2024
1 parent 0cf24ac commit 278a145
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 0 deletions.
16 changes: 16 additions & 0 deletions src/ATen/native/xpu/TensorProperties.cpp
Original file line number Diff line number Diff line change
@@ -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
1 change: 1 addition & 0 deletions yaml/xpu_functions.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -412,6 +412,7 @@ supported:
- _cdist_forward
- _pin_memory
- is_pinned
- is_set_to
- reflection_pad2d
- reflection_pad2d.out
- reflection_pad2d_backward
Expand Down

0 comments on commit 278a145

Please sign in to comment.