fix tests and Cmake

CMakeLists.txt

@@ -17,7 +17,7 @@ include(FetchContent)
 
 option(BUILD_TEST "Build C++ test binaries (need gtest and gbenchmark)" OFF)
 
-add_definitions("-D_GLIBCXX_USE_CXX11_ABI=0")
+add_definitions("-D_GLIBCXX_USE_CXX11_ABI=1")
 
 add_subdirectory(torchrec/csrc)
 

torchrec/ops/tests/faster_hash_bench.py

@@ -11,15 +11,22 @@
 import contextlib
 import logging
 import random
+import sys
 import time
 from typing import Any, Generator
 
 import torch
 
 logger: logging.Logger = logging.getLogger(__name__)
 
-torch.ops.load_library("//caffe2/torch/fb/retrieval:faster_hash_cpu")
-torch.ops.load_library("//caffe2/torch/fb/retrieval:faster_hash_cuda")
+def load_required_libraries() -> bool:
+    try:
+        torch.ops.load_library("//torchrec/ops:faster_hash_cpu")
+        torch.ops.load_library("//torchrec/ops:faster_hash_cuda")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to load faster_hash libraries, skipping test: {e}")
+        return False
 
 
 @contextlib.contextmanager
@@ -347,6 +354,9 @@ def _run_benchmark_with_eviction(
 
 
 if __name__ == "__main__":
+    if not load_required_libraries():
+        print("Skipping test because libraries were not loaded")
+        sys.exit(0)
     logger.setLevel(logging.INFO)
     handler = logging.StreamHandler()
     handler.setLevel(logging.INFO)

torchrec/ops/tests/faster_hash_test.py

@@ -13,16 +13,29 @@
 import torch
 from hypothesis import settings
 
-torch.ops.load_library("//torchrec/ops:faster_hash_cpu")
-torch.ops.load_library("//torchrec/ops:faster_hash_cuda")
-
+def load_required_libraries() -> bool:
+    try:
+        torch.ops.load_library("//torchrec/ops:faster_hash_cpu")
+        torch.ops.load_library("//torchrec/ops:faster_hash_cuda")
+        return True
+    except Exception as e:
+        print(f"Skipping tests because libraries were not loaded: {e}")
+        return False
 
 class HashZchKernelEvictionPolicy(IntEnum):
     THRESHOLD_EVICTION = 0
     LRU_EVICTION = 1
 
 
 class FasterHashTest(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        if not load_required_libraries():
+            raise unittest.SkipTest(
+                "Libraries not loaded, skipping all tests in MyTestCase"
+            )
+
     @unittest.skipIf(not torch.cuda.is_available(), "Skip when CUDA is not available")
     @settings(deadline=None)
     def test_simple_zch_no_evict(self) -> None: