Add some more tests for impl_abstract in other parts of test code (pytorch#2098)

Summary:

Sorry about the rapid fire PRs, I was expecting land to stop if OSS CI failed but it didn't lol

Differential Revision: D50696761

Author: ezyang

fbgemm_gpu/test/sparse_ops_test.py

@@ -17,6 +17,8 @@
 from itertools import accumulate
 from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union
 
+import fbgemm_gpu
+
 import hypothesis.strategies as st
 import numpy as np
 import torch
@@ -2424,7 +2426,11 @@ def validate(
 }
 
 # only generate tests on nightly pytorch (current release version is 2.1)
-if torch.__version__ >= "2.2.*":
+if (
+    torch.__version__ >= "2.2.*"
+    and hasattr(torch.library, "impl_abstract")
+    and not hasattr(fbgemm_gpu, "open_source")
+):
     generate_opcheck_tests(
         SparseOpsTest,
         ["fb", "fbgemm"],

fbgemm_gpu/test/test_utils.py

@@ -13,6 +13,8 @@
 from functools import wraps
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+import fbgemm_gpu
+
 import hypothesis.strategies as st
 import numpy as np
 import torch
@@ -204,7 +206,11 @@ def cpu_and_maybe_gpu() -> st.SearchStrategy[List[torch.device]]:
 
 
 def has_optests() -> bool:
-    return torch.__version__ >= "2.2.*"
+    return (
+        torch.__version__ >= "2.2.*"
+        and hasattr(torch.library, "impl_abstract")
+        and not hasattr(fbgemm_gpu, "open_source")
+    )
 
 
 class optests: