Update fori_loop.py

pytorch · Feb 29, 2024 · 216303b · 216303b
1 parent cd60e59
commit 216303b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py
@@ -26,11 +26,9 @@ def _xla_while_loop(cond_fn, body_fn, operands):
   shapes = xb.tensor_shape(operands)
   builder = xb.create_builder('test_while')
   params = []
-  # secondparams = []
   for shape in shapes:
     p = xb.mkparam(builder, len(params), shape)
     params.append(p)
-    # secondparams.append(xb.Op.tuple([p]))
 
   # generate cond_fn xlacomputation
   cond_result = cond_fn(operands)