From 21653dc9859c2fa2cf5e18ce79d7ac1e392c5381 Mon Sep 17 00:00:00 2001
From: JackCaoG <59073027+JackCaoG@users.noreply.github.com>
Date: Thu, 1 Feb 2024 17:57:03 -0800
Subject: [PATCH] Disable test_EmbeddingBag_per_sample_weights_and_new_offsets
 (#6456)

---
 test/pytorch_test_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/pytorch_test_base.py b/test/pytorch_test_base.py
index 4101d703d0d..64c8e41e4f4 100644
--- a/test/pytorch_test_base.py
+++ b/test/pytorch_test_base.py
@@ -19,7 +19,6 @@
     'test_sum_xla_bfloat16': 0.1,
     'test_put_xla_bfloat16': 0.05,
     'test_take_xla_bfloat16': 0.05,
-    'test_EmbeddingBag_per_sample_weights_and_new_offsets_xla': 0.01,
 }
 
 DISABLED_TORCH_TESTS_ANY = {
@@ -316,6 +315,7 @@
         'test_embedding_bag_device',  # FIXME! Unsupported device type for sparse layout: xla
         'test_embedding_scalar_weight_error_xla',  # tsl::CurrentStackTrace[abi:cxx11]
         'test_EmbeddingBag_per_sample_weights_and_no_offsets',  # FIXME! Unsupported device type for sparse layout: xla
+        'test_EmbeddingBag_per_sample_weights_and_new_offsets',  # precision
     },
 
     # test/nn/test_convolution.py