diff --git a/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py b/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py
index 4d0490dfdf..44915e38f5 100755
--- a/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py
+++ b/fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py
@@ -12,21 +12,7 @@
 import torch
 from torch import Tensor
 
-try:
-    try:
-        from torch.compiler import is_compiling
-
-        def is_torchdynamo_compiling() -> bool:  # type: ignore[misc]
-            # at least one test fails if we import is_compiling as a different name
-            return is_compiling()
-
-    except Exception:
-        # torch.compiler.is_compiling is not available in torch 1.10
-        from torch._dynamo import is_compiling as is_torchdynamo_compiling
-except Exception:
-
-    def is_torchdynamo_compiling() -> bool:  # type: ignore[misc]
-        return False
+from torch.compiler import is_compiling as is_torchdynamo_compiling
 
 
 # @manual=//deeplearning/fbgemm/fbgemm_gpu/codegen:split_embedding_codegen_lookup_invokers