Merge pull request #130 from jakevdp:nanobind

PiperOrigin-RevId: 592344892

Author: The ml_dtypes Authors

pyproject.toml

@@ -52,7 +52,6 @@ requires = [
     "numpy~=1.21.2; python_version<'3.11'",
     "numpy~=1.23.3; python_version>='3.11' and python_version<'3.12'",
     "numpy~=1.26.0; python_version>='3.12'",
-    "pybind11~=2.11.1",
     "setuptools~=68.1.0",
 ]
 build-backend = "setuptools.build_meta"

setup.py

@@ -17,19 +17,22 @@
 import fnmatch
 import platform
 import numpy as np
-from pybind11.setup_helpers import Pybind11Extension
+from setuptools import Extension
 from setuptools import setup
 from setuptools.command.build_py import build_py as build_py_orig
 
 if platform.system() == "Windows":
   COMPILE_ARGS = [
       "/std:c++17",
       "/DEIGEN_MPL2_ONLY",
+      "/EHsc",
+      "/bigobj",
   ]
 else:
   COMPILE_ARGS = [
       "-std=c++17",
       "-DEIGEN_MPL2_ONLY",
+      "-fvisibility=hidden",
       # -ftrapping-math is necessary because NumPy looks at floating point
       # exception state to determine whether to emit, e.g., invalid value
       # warnings. Without this setting, on Mac ARM we see spurious "invalid
@@ -56,7 +59,7 @@ def find_package_modules(self, package, package_dir):
 
 setup(
     ext_modules=[
-        Pybind11Extension(
+        Extension(
             "ml_dtypes._ml_dtypes_ext",
             [
                 "ml_dtypes/_src/dtypes.cc",