Skip to content

Commit

Permalink
Update jaten.py
Browse files Browse the repository at this point in the history
  • Loading branch information
chunnienc committed Sep 15, 2024
1 parent 2a5897d commit cf9f661
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 @@ -900,6 +900,7 @@ def _aten_relu(self):

@op(torch.ops.aten.cat)
def _aten_cat(tensors, dims=0):
tensors = [t for t in tensors if t.size > 0]
return jnp.concatenate(tensors, dims)


Expand Down

0 comments on commit cf9f661

Please sign in to comment.