Correct linking

mybigday · Oct 5, 2024 · 20953fc · 20953fc
1 parent b908b9f
commit 20953fc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/build.rs b/build.rs
@@ -74,9 +74,9 @@ fn main() {
         node_platform, node_arch
     );
 
-    assert!(link_lib(&lib_path, "executorch", false).is_ok());
-    if !link_lib(&lib_path, "executorch_no_prim_ops", false).is_ok() {
-        assert!(link_lib(&lib_path, "executorch_core", false).is_ok());
+    assert!(link_lib(&lib_path, "executorch", true).is_ok());
+    if !link_lib(&lib_path, "executorch_no_prim_ops", true).is_ok() {
+        assert!(link_lib(&lib_path, "executorch_core", true).is_ok());
     }
     if !link_lib(&lib_path, "extension_module_static", false).is_ok() {
         assert!(link_lib(&lib_path, "extension_module", false).is_ok());
@@ -99,6 +99,9 @@ fn main() {
         assert!(link_lib(&lib_path, "quantized_kernels", false).is_ok());
     }
 
+    // Custom Ops
+    let _ = link_lib(&lib_path, "custom_ops", true);
+
     // Tensor extension
     let _ = link_lib(&lib_path, "extension_tensor", false);