Revert D76866940: Decouple embedding_ssd_{}_pt2_autograd from CUDA files

Differential Revision:
D76866940

Original commit changeset: 2c09ae92cc0d

Original Phabricator Diff: D76866940

fbshipit-source-id: 4957c3958c927204c82221132831ace92c1fe2e7

Author: Wenbin Lin

fbgemm_gpu/fbgemm_gpu/utils/loader.py

@@ -14,10 +14,7 @@
 
 
 def load_torch_module(
-    unified_path: str,
-    cuda_path: Optional[str] = None,
-    hip_path: Optional[str] = None,
-    mtia_path: Optional[str] = None,
+    unified_path: str, cuda_path: Optional[str] = None, hip_path: Optional[str] = None
 ) -> None:
     try:
         torch.ops.load_library(unified_path)
@@ -27,16 +24,9 @@ def load_torch_module(
                 hip_path = f"{unified_path}_hip"
             torch.ops.load_library(hip_path)
         else:
-            try:
-                # pyre-ignore-next-line[21]
-                import mtia.host_runtime.torch_mtia.dynamic_library  # noqa
-
-                if mtia_path is not None:
-                    torch.ops.load_library(mtia_path)
-            except OSError:
-                if not cuda_path:
-                    cuda_path = f"{unified_path}_cuda"
-                torch.ops.load_library(cuda_path)
+            if not cuda_path:
+                cuda_path = f"{unified_path}_cuda"
+            torch.ops.load_library(cuda_path)
 
 
 def load_torch_module_bc(new_path: str, old_path: str) -> None: