Update jaten.py

pytorch · Sep 15, 2024 · 647a0bb · 647a0bb
1 parent cf9f661
commit 647a0bb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py
@@ -901,6 +901,8 @@ 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]
+  if not tensors:
+    return tensors[0]
   return jnp.concatenate(tensors, dims)