Fix flaky TBE unit tests (#3938)

Summary:

X-link: facebookresearch/FBGEMM#1023

As title

Reviewed By: spcyppt

Differential Revision: D72595672

Author: sryap

fbgemm_gpu/test/tbe/bench/tbe_data_config_models_test.py

@@ -18,7 +18,7 @@
     PoolingParams,
     TBEDataConfig,
 )
-from hypothesis import given
+from hypothesis import given, settings
 
 
 def rand_int(min_value: int, max_value: int) -> int:
@@ -30,6 +30,7 @@ class TBEDataConfigModelsTest(unittest.TestCase):
     @given(
         dtype=st.sampled_from([torch.float32, torch.float64]),
     )
+    @settings(deadline=None)
     def test_indices_params_serialization(
         self,
         dtype: torch.dtype,
@@ -57,6 +58,7 @@ def test_pooling_params_serialization(self) -> None:
     @given(
         dtype=st.sampled_from([torch.float32, torch.float64]),
     )
+    @settings(deadline=None)
     def test_tbe_data_config_serialization(
         self,
         dtype: torch.dtype,

fbgemm_gpu/test/tbe/training/failures_dict_fast.json

@@ -6,6 +6,7 @@
     "fbgemm::Fused8BitRowwiseQuantizedToFloat": {},
     "fbgemm::asynchronous_complete_cumsum": {},
     "fbgemm::bounds_check_indices": {},
+    "fbgemm::check_feature_gate_key": {},
     "fbgemm::dense_embedding_codegen_lookup_function": {
       "BackwardDenseTest.test_autograd_registration__test_backward_dense": {
         "comment": "",
@@ -24,6 +25,7 @@
       }
     },
     "fbgemm::emb_inplace_update": {},
+    "fbgemm::get_infos_metadata": {},
     "fbgemm::get_unique_indices": {
       "LXUCacheTest.test_faketensor__test_unique_lxu_cache_lookup": {
         "comment": "",
@@ -363,6 +365,7 @@
     "fbgemm::split_embedding_codegen_forward_weighted_cuda": {},
     "fbgemm::split_embedding_codegen_lookup_adagrad_function": {},
     "fbgemm::split_embedding_codegen_lookup_adagrad_function_cpu": {},
+    "fbgemm::split_embedding_codegen_lookup_adagrad_function_pt2": {},
     "fbgemm::split_embedding_codegen_lookup_adam_function": {},
     "fbgemm::split_embedding_codegen_lookup_lamb_function": {},
     "fbgemm::split_embedding_codegen_lookup_lars_sgd_function": {},
@@ -381,10 +384,6 @@
     },
     "fbgemm::split_embedding_codegen_lookup_rowwise_adagrad_function_cpu": {},
     "fbgemm::split_embedding_codegen_lookup_rowwise_adagrad_function_pt2": {
-      "BackwardAdagradTest.test_faketensor__test_backward_adagrad_fp32_cpu": {
-        "comment": "",
-        "status": "xfail"
-      },
       "ForwardTest.test_faketensor__test_forward_cpu_fp32": {
         "comment": "",
         "status": "xfail"