copy is_non_strict_exporting over to resolve backward compatibility…

… issue

Summary:
# context
* An ImportError was reported
```
ImportError: cannot import name 'register_custom_op' from 'torchrec.modules.utils'
```
* it's due to previously introduced function at D56443608
* test failed: https://www.internalfb.com/intern/test/562950051370255/
```
Mitigation: ImportError("cannot import name 'is_non_strict_exporting' from '<torch_package_1>.torchrec.sparse.jagged_tensor' (<torch_package_1>.torchrec/sparse/jagged_tensor.py)")
```

Reviewed By: IvanKobzarev, drdarshan

Differential Revision: D56578871

fbshipit-source-id: 3f447cac2946ac8b270848c0067bf990f3e31707

torchrec/modules/embedding_modules.py

@@ -19,12 +19,25 @@
     pooling_type_to_str,
 )
 from torchrec.modules.utils import register_custom_op
-from torchrec.sparse.jagged_tensor import (
-    is_non_strict_exporting,
-    JaggedTensor,
-    KeyedJaggedTensor,
-    KeyedTensor,
-)
+from torchrec.sparse.jagged_tensor import JaggedTensor, KeyedJaggedTensor, KeyedTensor
+
+
+try:
+    if torch.jit.is_scripting():
+        raise Exception()
+
+    from torch.compiler import (
+        is_compiling as is_compiler_compiling,
+        is_dynamo_compiling as is_torchdynamo_compiling,
+    )
+
+    def is_non_strict_exporting() -> bool:
+        return not is_torchdynamo_compiling() and is_compiler_compiling()
+
+except Exception:
+
+    def is_non_strict_exporting() -> bool:
+        return False
 
 
 @torch.fx.wrap