add quantize ops to pybindings

lucylq · lucylq · commit cd9d73084065 · 2024-04-22T10:22:16.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
@@ -376,6 +376,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
`@@ -376,6 +376,7 @@ def run(self):`
`376`	`376`	`if ShouldBuild.pybindings:`
`377`	`377`	`cmake_args += [`
`378`	`378`	`"-DEXECUTORCH_BUILD_PYBIND=ON",`
	`379`	`+ "-DEXECUTORCH_BUILD_QUANTIZED=ON", # add quantized ops to pybindings.`
`379`	`380`	`]`
`380`	`381`	`build_args += ["--target", "portable_lib"]`
`381`	`382`	`if ShouldBuild.xnnpack:`