diff --git a/setup.py b/setup.py
index 5718687..850f83d 100644
--- 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",
+    # ],
 )