Skip to content

Commit

Permalink
Remove reshape lowering (#6230)
Browse files Browse the repository at this point in the history
  • Loading branch information
JackCaoG authored and golechwierowicz committed Jan 12, 2024
1 parent c92ac00 commit 6cfaea8
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 13 deletions.
3 changes: 0 additions & 3 deletions codegen/xla_native_functions.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -371,7 +371,6 @@ supported:
- narrow_copy
- pixel_shuffle
- pixel_unshuffle
- reshape
- select_backward
- select.int
- slice.Tensor
Expand Down Expand Up @@ -404,8 +403,6 @@ symint:
- narrow_copy
- select_backward
- select.int
# See Note: [functionalization and CompositeExplicitAutograd]
- reshape
# See Note: [Disabling functionalization]
- expand
- view
Expand Down
10 changes: 0 additions & 10 deletions torch_xla/csrc/aten_xla_type.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -3578,16 +3578,6 @@ at::Tensor XLANativeFunctions::pixel_unshuffle(const at::Tensor& self,
pixel_unshuffle)>::call(self, downscale_factor);
}

at::Tensor XLANativeFunctions::reshape_symint(const at::Tensor& self,
c10::SymIntArrayRef shape) {
// See Note: [Disabling functionalization]
if (runtime::sys_util::GetEnvBool("XLA_DISABLE_FUNCTIONALIZATION", false)) {
return at::native::reshape_symint(self, shape);
}
return at::functionalization::functionalize_aten_op_symint<ATEN_OP(
reshape)>::call(self, shape);
}

at::Tensor XLANativeFunctions::select_backward_symint(
const at::Tensor& grad_output, c10::SymIntArrayRef input_sizes, int64_t dim,
c10::SymInt index) {
Expand Down

0 comments on commit 6cfaea8

Please sign in to comment.