Use lib paths working with editable mode

Author: AdrianLundell

backends/arm/test/conftest.py

@@ -195,9 +195,9 @@ def _load_lib(lib_name_pattern: str, build_folder: str):
         )
 
 
-def _load_libquantized_ops_aot_lib(executorch_install_dir: str = "pip-out"):
+def _load_libquantized_ops_aot_lib():
     """
     Find and load the libquantized_ops_aot_lib shared library.
     """
-    _load_lib("_portable_lib.cpython-310*", executorch_install_dir)
-    _load_lib("libquantized_ops_aot_lib.*", executorch_install_dir)
+    _load_lib("_portable_lib.cpython-310*", "extension")
+    _load_lib("libquantized_ops_aot_lib.*", "kernels")

examples/arm/aot_arm_compiler.py

@@ -513,7 +513,7 @@ def get_args():
     # Load quantized ops library.
     if args.quantize:
         logging.info("Loading lib_quantized_custom_op_lib")
-        _load_libquantized_ops_aot_lib(executorch_install_dir="pip-out")
+        _load_libquantized_ops_aot_lib()
 
     # if we have custom ops, register them before processing the model
     if args.so_library is not None: