Disable dynamo testing in TBE

sryap · facebook-github-bot · commit 89d661d39807 · 2024-03-02T15:48:42.000-08:00
Summary: This diff disables dynamo testing in TBE unit tests.

Differential Revision: D54455966
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:

Original file line number	Diff line number	Diff line change
`@@ -119,6 +119,9 @@ def execute_backward_adagrad_( # noqa C901`
`119`	`119`	`)`
`120`	`120`	`)`
`121`	`121`
	`122`	`+ # Disable dynamo tests due to unknown failure`
	`123`	`+ assume(not compile)`
	`124`	`+`
`122`	`125`	`emb_op = SplitTableBatchedEmbeddingBagsCodegen`
`123`	`126`
`124`	`127`	`if pooling_mode == PoolingMode.SUM:`