Skip to content

Commit

Permalink
Add "# aten.unique_consecutive" comment above impl
Browse files Browse the repository at this point in the history
Most of the other lowerings have similar comments, so this is just for
consistency.
  • Loading branch information
mrguenther committed Oct 16, 2024
1 parent 0070559 commit 625642a
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions experimental/torch_xla2/torch_xla2/ops/jaten.py
Original file line number Diff line number Diff line change
Expand Up @@ -2775,6 +2775,7 @@ def _aten_unique2(input_tensor,
return_counts=return_counts)


# aten.unique_consecutive
@op(torch.ops.aten.unique_consecutive)
def _aten_unique_consecutive(input_tensor,
return_inverse=False,
Expand Down

0 comments on commit 625642a

Please sign in to comment.