Skip to content

Commit

Permalink
Re-organize TBE tests, pt 3 (pytorch#2272)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#2272

- Re-organize TBE tests, pt 3

Reviewed By: spcyppt

Differential Revision: D52857686

fbshipit-source-id: 756cdf9b24bb80754e4a482b1e4e63f96633331a
  • Loading branch information
q10 authored and facebook-github-bot committed Jan 19, 2024
1 parent 9a3c5b2 commit 5b3e085
Show file tree
Hide file tree
Showing 8 changed files with 2,797 additions and 2,629 deletions.
506 changes: 506 additions & 0 deletions fbgemm_gpu/test/tbe/cache_test.py

Large diffs are not rendered by default.

166 changes: 31 additions & 135 deletions fbgemm_gpu/test/tbe/failures_dict_fast.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,54 +16,42 @@
},
"fbgemm::direct_mapped_lru_cache_populate_byte": {},
"fbgemm::direct_mapped_lxu_cache_lookup": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_direct_mapped_uvm_cache_stats": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
}
},
"fbgemm::emb_inplace_update": {},
"fbgemm::get_unique_indices": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_unique_lxu_cache_lookup": {
"LXUCacheTest.test_faketensor__test_unique_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
}
},
"fbgemm::int_nbit_split_embedding_codegen_lookup_function": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xsuccess"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_cache_miss_counter": {
"NBitCacheTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xsuccess"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_direct_mapped_uvm_cache_stats": {
"NBitForwardTest.test_faketensor__test_nbit_forward_fused_pooled_emb_quant": {
"comment": "",
"status": "xsuccess"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_fused_pooled_emb_quant": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xsuccess"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xsuccess"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_uvm_cache_stats": {
"NBitSplitEmbeddingTest.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": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"NBitSplitEmbeddingTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"NBitSplitEmbeddingTest.test_schema__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
}
Expand All @@ -76,18 +64,10 @@
"BackwardSGDTest.test_faketensor__test_backward_sgd_really_long_segments": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_pipeline": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_cache_pipeline": {
"comment": "",
"status": "skip"
}
},
"fbgemm::lfu_cache_populate_byte": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
}
Expand Down Expand Up @@ -125,61 +105,37 @@
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_miss_counter": {
"CacheTest.test_faketensor__test_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_pipeline": {
"LXUCacheTest.test_faketensor__test_unique_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline": {
"LinearizeCacheIndicesTest.test_faketensor__test_linearize_cache_indices": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_1": {
"NBitCacheTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_2": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_linearize_cache_indices": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_direct_mapped_uvm_cache_stats": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_uvm_cache_stats": {
"NBitSplitEmbeddingTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_stb_uvm_cache_stats": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_unique_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
}
},
"fbgemm::linearize_cache_indices_from_row_idx": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_linearize_cache_indices_from_row_idx": {
"LinearizeCacheIndicesTest.test_faketensor__test_linearize_cache_indices_from_row_idx": {
"comment": "",
"status": "xfail"
}
Expand Down Expand Up @@ -217,45 +173,13 @@
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_miss_counter": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_pipeline": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_1": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_2": {
"CacheTest.test_faketensor__test_cache_miss_counter": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_stb_uvm_cache_stats": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_cache_pipeline": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_cache_prefetch_pipeline": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_cache_prefetch_pipeline_stream_1": {
"comment": "",
"status": "skip"
},
"SplitTableBatchedEmbeddingsTest.test_schema__test_cache_prefetch_pipeline_stream_2": {
"comment": "",
"status": "skip"
}
},
"fbgemm::lru_cache_populate_byte": {},
Expand Down Expand Up @@ -294,92 +218,64 @@
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_miss_counter": {
"CacheTest.test_faketensor__test_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_pipeline": {
"LXUCacheTest.test_faketensor__test_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline": {
"LXUCacheTest.test_faketensor__test_unique_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_1": {
"NBitCacheTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_cache_prefetch_pipeline_stream_2": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_uvm_cache_stats": {
"NBitSplitEmbeddingTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_stb_uvm_cache_stats": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_unique_lxu_cache_lookup": {
"comment": "",
"status": "xfail"
}
},
"fbgemm::new_managed_tensor": {},
"fbgemm::new_unified_tensor": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_cache_update_function": {
"NBitCacheTest.test_faketensor__test_nbit_cache_miss_counter": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_direct_mapped_uvm_cache_stats": {
"NBitCacheTest.test_faketensor__test_nbit_cache_update_function": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_gpu_no_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_gpu_no_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_gpu_no_cache_fp8_2048": {
"NBitForwardTest.test_faketensor__test_nbit_forward_gpu_no_cache_fp8_2048": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_uvm_cache_stats": {
"NBitSplitEmbeddingTest.test_faketensor__test_int_nbit_split_embedding_uvm_caching_codegen_lookup_function": {
"comment": "",
"status": "xfail"
}
},
"fbgemm::pruned_array_lookup": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
Expand All @@ -390,7 +286,7 @@
},
"fbgemm::pruned_hashmap_insert": {},
"fbgemm::pruned_hashmap_lookup": {
"SplitTableBatchedEmbeddingsTest.test_faketensor__test_nbit_forward_uvm_cache": {
"NBitForwardTest.test_faketensor__test_nbit_forward_uvm_cache": {
"comment": "",
"status": "xfail"
},
Expand Down
Loading

0 comments on commit 5b3e085

Please sign in to comment.