From 57cce80357afe017e410c8cd39825de2da568e49 Mon Sep 17 00:00:00 2001 From: Simon Teo Date: Tue, 5 Nov 2024 02:13:15 +0800 Subject: [PATCH] Remove unnecessary functions and print statements (#7495) (#8352) Co-authored-by: Simon Teo --- experimental/torch_xla2/torch_xla2/ops/jaten.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py index f1c959c2be6..08f27de7940 100644 --- a/experimental/torch_xla2/torch_xla2/ops/jaten.py +++ b/experimental/torch_xla2/torch_xla2/ops/jaten.py @@ -624,12 +624,6 @@ def _zeros(size: Sequence[int], dtype=None, **kwargs): return jnp.zeros(size, dtype) -@op(torch.ops.aten.eye) -@op_base.convert_dtype() -def _eye(n: int, m: Optional[int] = None, *, dtype=None, **kwargs): - return jnp.eye(n, m, dtype=dtype) - - @op(torch.ops.aten.full) @op_base.convert_dtype() def _full(size: Sequence[int], fill_value, *, dtype=None, **kwargs): @@ -1892,7 +1886,6 @@ def _helper(curr_dim_idx, sofar): _helper(curr_dim_idx + 1, sofar[:]) return if curr_dim_idx >= len(dims): - print(sofar) res.append(sofar) return for i in range(dims[curr_dim_idx]):