Disable test_EmbeddingBag_per_sample_weights_and_new_offsets (#6456)

pytorch · Feb 2, 2024 · 21653dc · 21653dc
1 parent d1a454f
commit 21653dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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