Fix ABI incompatibility with pybind11

HolyWu · HolyWu · commit 83a23800be36 · 2024-05-18T09:49:27.000+08:00
diff --git a/setup.py b/setup.py
@@ -15,6 +15,7 @@
 from typing import List
 
 import setuptools
+import torch
 import yaml
 from setuptools import Extension, find_namespace_packages, setup
 from setuptools.command.build_ext import build_ext
@@ -504,6 +505,7 @@ def run(self):
             ],
             extra_compile_args=(
                 [
+                    f'/DPYBIND11_BUILD_ABI=\\"{torch._C._PYBIND11_BUILD_ABI}\\"',
                     "/GS-",
                     "/permissive-",
                 ]

Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,7 @@`
`15`	`15`	`from typing import List`
`16`	`16`
`17`	`17`	`import setuptools`
	`18`	`+import torch`
`18`	`19`	`import yaml`
`19`	`20`	`from setuptools import Extension, find_namespace_packages, setup`
`20`	`21`	`from setuptools.command.build_ext import build_ext`
`@@ -504,6 +505,7 @@ def run(self):`
`504`	`505`	`],`
`505`	`506`	`extra_compile_args=(`
`506`	`507`	`[`
	`508`	`+ f'/DPYBIND11_BUILD_ABI=\\"{torch._C._PYBIND11_BUILD_ABI}\\"',`
`507`	`509`	`"/GS-",`
`508`	`510`	`"/permissive-",`
`509`	`511`	`]`