add quantize ops to pybindings

lucylq · lucylq · commit 913a755e7d9d · 2024-04-22T10:20:44.000-07:00
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -562,6 +562,11 @@ if(EXECUTORCH_BUILD_PYBIND)
     list(APPEND _dep_libs custom_ops)
   endif()
 
+  if(EXECUTORCH_BUILD_QUANTIZED)
+    target_link_options_shared_lib(quantized_ops_lib)
+    list(APPEND _dep_libs quantized_kernels quantized_ops_lib)
+  endif()
+
   # TODO(larryliu): Fix macOS 2 dylibs having 2 sets of static variables issue
   if(EXECUTORCH_BUILD_CUSTOM_OPS_AOT AND NOT APPLE)
     list(APPEND _dep_libs custom_ops_aot_lib)
diff --git a/setup.py b/setup.py
@@ -375,6 +375,7 @@ def run(self):
         if ShouldBuild.pybindings:
             cmake_args += [
                 "-DEXECUTORCH_BUILD_PYBIND=ON",
+                "-DEXECUTORCH_BUILD_QUANTIZED=ON",  # add quantized ops to pybindings.
             ]
             build_args += ["--target", "portable_lib"]
             if ShouldBuild.xnnpack:

Original file line number	Diff line number	Diff line change
`@@ -375,6 +375,7 @@ def run(self):`
`375`	`375`	`if ShouldBuild.pybindings:`
`376`	`376`	`cmake_args += [`
`377`	`377`	`"-DEXECUTORCH_BUILD_PYBIND=ON",`
	`378`	`+ "-DEXECUTORCH_BUILD_QUANTIZED=ON", # add quantized ops to pybindings.`
`378`	`379`	`]`
`379`	`380`	`build_args += ["--target", "portable_lib"]`
`380`	`381`	`if ShouldBuild.xnnpack:`