From 3d6d27be2a910f85b0112bf6dad71996932f3930 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Wed, 20 Nov 2024 14:07:22 -0800
Subject: [PATCH] Update test_ops.py

---
 experimental/torch_xla2/test/test_ops.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/experimental/torch_xla2/test/test_ops.py b/experimental/torch_xla2/test/test_ops.py
index 28d0f29f0c1..848c55733e9 100644
--- a/experimental/torch_xla2/test/test_ops.py
+++ b/experimental/torch_xla2/test/test_ops.py
@@ -35,9 +35,6 @@
     "nn.functional.conv_transpose2d",
     "nn.functional.conv_transpose3d",
     "nn.functional.ctc_loss",
-    "nn.functional.dropout2d",
-    "nn.functional.dropout3d",
-    "nn.functional.dropout",
     "nn.functional.embedding_bag",
     "nn.functional.fractional_max_pool2d",
     "nn.functional.fractional_max_pool3d",
@@ -98,6 +95,9 @@
   'exponential',
   'log_normal',
   'randint',
+  'nn.functional.dropout2d',
+  'nn.functional.dropout3d',
+  'nn.functional.dropout',
 }
 
 atol_dict = {"linalg.eig": (2e0, 3e0),