From 9f1d0ef86d26f61e708b4e0ad851b004e347158e Mon Sep 17 00:00:00 2001 From: Sarunya Pumma Date: Mon, 27 May 2024 18:33:46 -0700 Subject: [PATCH] Add TBE tests to failures_dict_fast.json (#2632) Summary: Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/2632 As title Reviewed By: q10, spcyppt Differential Revision: D57758099 fbshipit-source-id: d90aaa5ce6461251e53f08074504b4cc2738eaab --- .../test/tbe/cache/failures_dict_fast.json | 4 ++ .../tbe/inference/failures_dict_fast.json | 14 +++--- .../test/tbe/training/failures_dict_fast.json | 47 +++++++++++++++++++ .../test/tbe/utils/failures_dict_fast.json | 4 ++ 4 files changed, 61 insertions(+), 8 deletions(-) diff --git a/fbgemm_gpu/test/tbe/cache/failures_dict_fast.json b/fbgemm_gpu/test/tbe/cache/failures_dict_fast.json index 6e5930718..5feb843e6 100644 --- 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 index ec2f87df2..00360f9f4 100644 --- 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,10 +35,6 @@ } }, "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" @@ -40,10 +42,6 @@ "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 index 47c1a36ca..e419a1a34 100644 --- 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 index 6e5930718..5864eb5c4 100644 --- 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": {},