Add TBE tests to failures_dict_fast.json (#2632)

Summary: Pull Request resolved: #2632 As title Reviewed By: q10, spcyppt Differential Revision: D57758099 fbshipit-source-id: d90aaa5ce6461251e53f08074504b4cc2738eaab
pytorch · May 28, 2024 · 9f1d0ef · 9f1d0ef
1 parent 5dd515e
commit 9f1d0ef
Show file tree

Hide file tree

Showing 4 changed files with 61 additions and 8 deletions.
diff --git a/fbgemm_gpu/test/tbe/cache/failures_dict_fast.json b/fbgemm_gpu/test/tbe/cache/failures_dict_fast.json
@@ -222,6 +222,10 @@
         "comment": "",
         "status": "xfail"
       },
+      "CacheTest.test_schema__test_cache_miss_counter": {
+        "comment": "",
+        "status": "xfail"
+      },
       "LXUCacheTest.test_faketensor__test_lxu_cache_lookup": {
         "comment": "",
         "status": "xfail"

diff --git a/fbgemm_gpu/test/tbe/inference/failures_dict_fast.json b/fbgemm_gpu/test/tbe/inference/failures_dict_fast.json
@@ -2,6 +2,12 @@
   "_description": "This is a dict containing failures for tests autogenerated by generate_opcheck_tests. For more details, please see https://docs.google.com/document/d/1Pj5HRZvdOq3xpFpbEjUZp2hBovhy7Wnxw14m6lF2154/edit",
   "_version": 1,
   "data": {
+    "fbgemm::FloatToFused8BitRowwiseQuantized": {},
+    "fbgemm::FloatToFusedNBitRowwiseQuantizedSBHalf": {},
+    "fbgemm::FloatToHFP8Quantized": {},
+    "fbgemm::Fused8BitRowwiseQuantizedToFloat": {},
+    "fbgemm::Fused8BitRowwiseQuantizedToFloatOrHalf": {},
+    "fbgemm::HFP8QuantizedToFloat": {},
     "fbgemm::asynchronous_complete_cumsum": {},
     "fbgemm::bounds_check_indices": {},
     "fbgemm::dense_embedding_codegen_lookup_function": {
@@ -29,21 +35,13 @@
       }
     },
     "fbgemm::int_nbit_split_embedding_codegen_lookup_function": {
-      "NBitCacheTest.test_faketensor__test_nbit_cache_miss_counter": {
-        "comment": "",
-        "status": "xsuccess"
-      },
       "NBitForwardTest.test_faketensor__test_nbit_forward_fused_pooled_emb_quant": {
         "comment": "",
         "status": "xsuccess"
       },
       "NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
         "comment": "",
         "status": "xsuccess"
-      },
-      "NBitSplitEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
-        "comment": "",
-        "status": "xsuccess"
       }
     },
     "fbgemm::int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {

diff --git a/fbgemm_gpu/test/tbe/training/failures_dict_fast.json b/fbgemm_gpu/test/tbe/training/failures_dict_fast.json
@@ -2,6 +2,8 @@
   "_description": "This is a dict containing failures for tests autogenerated by generate_opcheck_tests. For more details, please see https://docs.google.com/document/d/1Pj5HRZvdOq3xpFpbEjUZp2hBovhy7Wnxw14m6lF2154/edit",
   "_version": 1,
   "data": {
+    "fbgemm::FloatToFused8BitRowwiseQuantized": {},
+    "fbgemm::Fused8BitRowwiseQuantizedToFloat": {},
     "fbgemm::asynchronous_complete_cumsum": {},
     "fbgemm::bounds_check_indices": {},
     "fbgemm::dense_embedding_codegen_lookup_function": {
@@ -198,6 +200,10 @@
         "comment": "",
         "status": "xfail"
       },
+      "BackwardAdagradLargeDimTest.test_schema__test_backward_adagrad_large_dims": {
+        "comment": "",
+        "status": "xfail"
+      },
       "BackwardAdagradTest.test_faketensor__test_backward_adagrad_fp16_pmMEAN": {
         "comment": "",
         "status": "xfail"
@@ -226,6 +232,30 @@
         "comment": "",
         "status": "xfail"
       },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp16_pmMEAN": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp16_pmNONE": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp16_pmSUM": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp16_pmSUM_with_max_norm": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp32_pmNONE": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardAdagradTest.test_schema__test_backward_adagrad_fp32_pmSUM": {
+        "comment": "",
+        "status": "xfail"
+      },
       "BackwardSGDTest.test_faketensor__test_backward_sgd": {
         "comment": "",
         "status": "xfail"
@@ -234,10 +264,22 @@
         "comment": "",
         "status": "xfail"
       },
+      "BackwardSGDTest.test_schema__test_backward_sgd": {
+        "comment": "",
+        "status": "xfail"
+      },
+      "BackwardSGDTest.test_schema__test_backward_sgd_really_long_segments": {
+        "comment": "",
+        "status": "xfail"
+      },
       "CacheTest.test_faketensor__test_cache_miss_counter": {
         "comment": "",
         "status": "xfail"
       },
+      "CacheTest.test_schema__test_cache_miss_counter": {
+        "comment": "",
+        "status": "xfail"
+      },
       "LXUCacheTest.test_faketensor__test_lxu_cache_lookup": {
         "comment": "",
         "status": "xfail"
@@ -261,6 +303,10 @@
       "SplitTableBatchedEmbeddingsTest.test_faketensor__test_stb_uvm_cache_stats": {
         "comment": "",
         "status": "xfail"
+      },
+      "SplitTableBatchedEmbeddingsTest.test_schema__test_stb_uvm_cache_stats": {
+        "comment": "",
+        "status": "xfail"
       }
     },
     "fbgemm::new_managed_tensor": {},
@@ -325,6 +371,7 @@
     "fbgemm::split_embedding_codegen_lookup_partial_rowwise_lamb_function": {},
     "fbgemm::split_embedding_codegen_lookup_rowwise_adagrad_function": {},
     "fbgemm::split_embedding_codegen_lookup_rowwise_adagrad_function_cpu": {},
+    "fbgemm::split_embedding_codegen_lookup_rowwise_adagrad_with_counter_function": {},
     "fbgemm::split_embedding_codegen_lookup_rowwise_weighted_adagrad_function": {},
     "fbgemm::split_embedding_codegen_lookup_sgd_function": {},
     "fbgemm::split_embedding_codegen_lookup_sgd_function_cpu": {}

diff --git a/fbgemm_gpu/test/tbe/utils/failures_dict_fast.json b/fbgemm_gpu/test/tbe/utils/failures_dict_fast.json
@@ -245,6 +245,10 @@
       "SplitTableBatchedEmbeddingsTest.test_faketensor__test_stb_uvm_cache_stats": {
         "comment": "",
         "status": "xfail"
+      },
+      "SplitTableBatchedEmbeddingsTest.test_schema__test_stb_uvm_cache_stats": {
+        "comment": "",
+        "status": "xfail"
       }
     },
     "fbgemm::new_managed_tensor": {},