Fix SSD TBE tests for OSS (pytorch#2665)

Summary:
- Fix SSD TBE test for OSS

Pull Request resolved: pytorch#2665

Reviewed By: sryap, spcyppt

Differential Revision: D58099204

Pulled By: q10

fbshipit-source-id: 5705c8d0021392189c8aa41978f5bd387c70a05c

.github/scripts/fbgemm_gpu_test.bash

@@ -66,7 +66,7 @@ run_python_test () {
 
 __configure_fbgemm_gpu_test_cpu () {
   ignored_tests=(
-    ./ssd_split_table_batched_embeddings_test.py
+    ./tbe/ssd/ssd_split_table_batched_embeddings_test.py
     # These tests have non-CPU operators referenced in @given
     ./uvm/copy_test.py
     ./uvm/uvm_test.py
@@ -77,7 +77,7 @@ __configure_fbgemm_gpu_test_cpu () {
 
 __configure_fbgemm_gpu_test_cuda () {
   ignored_tests=(
-    ./ssd_split_table_batched_embeddings_test.py
+    ./tbe/ssd/ssd_split_table_batched_embeddings_test.py
     # require multiple GPUs
     ./comm/multi_gpu_car_test.py
   )
@@ -102,7 +102,7 @@ __configure_fbgemm_gpu_test_rocm () {
   fi
 
   ignored_tests=(
-    ./ssd_split_table_batched_embeddings_test.py
+    ./tbe/ssd/ssd_split_table_batched_embeddings_test.py
     # https://github.com/pytorch/FBGEMM/issues/1559
     ./batched_unary_embeddings_test.py
     # require multiple GPUs

fbgemm_gpu/fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py

@@ -48,14 +48,18 @@
             "//deeplearning/fbgemm/fbgemm_gpu:ssd_split_table_batched_embeddings_hip"
         )
     else:
-        torch.ops.load_library(
-            "//deeplearning/fbgemm/fbgemm_gpu:ssd_split_table_batched_embeddings"
-        )
-except OSError:
-    # Keep for BC: will be deprecated soon.
-    torch.ops.load_library(
-        "//deeplearning/fbgemm/fbgemm_gpu/fb:ssd_split_table_batched_embeddings"
-    )
+        try:
+            torch.ops.load_library(
+                "//deeplearning/fbgemm/fbgemm_gpu:ssd_split_table_batched_embeddings"
+            )
+        except OSError:
+            # Keep for BC: will be deprecated soon.
+            torch.ops.load_library(
+                "//deeplearning/fbgemm/fbgemm_gpu/fb:ssd_split_table_batched_embeddings"
+            )
+except Exception:
+    pass
+
 
 ASSOC = 32