Disable dynamo testing in TBE (pytorch#2381)

Summary: Pull Request resolved: pytorch#2381 This diff disables dynamo testing in TBE unit tests. Reviewed By: xing-liu Differential Revision: D54455966 fbshipit-source-id: 9e9c7b0fc40e850f79d9a8d6cf6e1b892f6e4db8
q10 · Mar 4, 2024 · e764e6f · e764e6f
1 parent f072386
commit e764e6f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fbgemm_gpu/test/tbe/training/backward_adagrad_test.py b/fbgemm_gpu/test/tbe/training/backward_adagrad_test.py
@@ -119,6 +119,9 @@ def execute_backward_adagrad_(  # noqa C901
             )
         )
 
+        # Disable dynamo tests due to unknown failure
+        assume(not compile)
+
         emb_op = SplitTableBatchedEmbeddingBagsCodegen
 
         if pooling_mode == PoolingMode.SUM: