diff --git a/test/test_test_while_loop_xla_computation.py b/test/test_test_while_loop_xla_computation.py deleted file mode 100644 index 77c9c0eb054..00000000000 --- a/test/test_test_while_loop_xla_computation.py +++ /dev/null @@ -1,25 +0,0 @@ -# import time -# start_time = time.time() -import torch -import torch_xla -import torch_xla.experimental.fori_loop -from torch._higher_order_ops.while_loop import while_loop -import torch_xla.core.xla_model as xm -import torch_xla.core.xla_builder as xb -# import torch_xla.debug.profiler as xp - -device = xm.xla_device() - -def cond_fn(init, limit_value): - return limit_value[0] >= init[0] - -def body_fn(init, limit_value): - one_value = torch.ones(1, dtype=torch.int32, device=device) - return (torch.add(init, one_value), limit_value.clone()) - -init = torch.tensor([0], dtype=torch.int32, device=device) -limit_value = torch.tensor([100], dtype=torch.int32, device=device) -res = while_loop(cond_fn, body_fn, (init, limit_value)) -print("res: ", res) - -print("--- %s seconds ---" % (time.time() - start_time)) \ No newline at end of file