update version compatibility to allow torch 2.11.0.dev

stack-info: PR: #3545, branch: danielvegamyhre/stack/91

Author: danielvegamyhre

setup.py

@@ -722,6 +722,7 @@ def get_extensions():
                     include_dirs=[
                         mxfp8_extension_dir,  # For mxfp8_quantize.cuh, mxfp8_extension.cpp, and mxfp8_cuda.cu
                     ],
+                    libraries=["cuda"],
                     extra_compile_args={
                         "cxx": [
                             f"-DPy_LIMITED_API={min_supported_cpython_hexcode}",

torchao/__init__.py

@@ -70,9 +70,7 @@ def _parse_version(version_string):
     torchao_pytorch_compatible_versions = [
         # Built against torch 2.8.0
         (_parse_version("0.13.0"), _parse_version("2.8.0")),
-        (_parse_version("0.13.0"), _parse_version("2.9.0.dev")),
         (_parse_version("0.14.0"), _parse_version("2.8.0")),
-        (_parse_version("0.14.0"), _parse_version("2.9.0.dev")),
         # Built against torch 2.9.0
         (_parse_version("0.14.1"), _parse_version("2.9.0")),
         (_parse_version("0.14.1"), _parse_version("2.10.0.dev")),
@@ -82,6 +80,7 @@ def _parse_version(version_string):
         # Current torchao version
         (_parse_version("0.16.0.dev"), _parse_version("2.9.1")),
         (_parse_version("0.16.0.dev"), _parse_version("2.10.0.dev")),
+        (_parse_version("0.16.0.dev"), _parse_version("2.11.0.dev")),
     ]
 
     current_torch_version = _parse_version(torch.__version__)
@@ -108,10 +107,15 @@ def _parse_version(version_string):
     try:
         from pathlib import Path
 
+        # Load abi3 .so files and cpython .so files
         so_files = list(Path(__file__).parent.glob("_C*.so"))
         if len(so_files) > 0:
             for file in so_files:
-                torch.ops.load_library(str(file))
+                logger.debug(f"Loading {file}")
+                try:
+                    torch.ops.load_library(str(file))
+                except Exception as e:
+                    logger.debug(f"Failed to load {file}: {e}")
             from . import ops
 
         # The following registers meta kernels for some CPU kernels