include cpp extension in setup.py

DiffAPF · Jan 22, 2025 · d77846a · d77846a
1 parent 1f392ba
commit d77846a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 import setuptools
+from torch.utils import cpp_extension
 
 NAME = "torchlpc"
 VERSION = "0.6"
@@ -25,4 +26,13 @@
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
+    ext_modules=[
+        cpp_extension.CppExtension("torchlpc", ["torchlpc/csrc/scan_cpu.cpp"])
+    ],
+    cmdclass={"build_ext": cpp_extension.BuildExtension},
+    # include_dirs=[
+    #     "/Library/Developer/CommandLineTools/usr/lib/clang/16/include",
+    #     "/Library/Developer/CommandLineTools/usr/include",
+    #     "/Library/Developer/CommandLineTools/SDKs/MacOSX15.2.sdk/usr/include",
+    # ],
 )