Fix ABI incompatibility with pybind11

HolyWu · HolyWu · commit f8f0ab755db2 · 2024-05-08T01:01:07.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
@@ -508,6 +509,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`
`@@ -508,6 +509,7 @@ def run(self):`
`508`	`509`	`],`
`509`	`510`	`extra_compile_args=(`
`510`	`511`	`[`
	`512`	`+ f'/DPYBIND11_BUILD_ABI=\\"{torch._C._PYBIND11_BUILD_ABI}\\"',`
`511`	`513`	`"/GS-",`
`512`	`514`	`"/permissive-",`
`513`	`515`	`]`