adaptor for moving ark.so to auto_round_exteion.ark

Author: Zhenzhong1

auto_round/inference/backend.py

@@ -631,7 +631,7 @@ def dynamic_import_inference_linear(backend, config):
 
     if "auto_round_kernel" in backend or "ark" in backend:
         try:
-            import auto_round_kernel as ark  # pylint: disable=E0401
+            from auto_round_extension.ark import auto_round_kernel as ark  # pylint: disable=E0401
         except Exception as e:
             raise ImportError("Please install auto_round_kernel version for CPU/XPU")
         import auto_round_extension.ark.qlinear as qlinear

auto_round_extension/ark/qlinear.py

@@ -20,7 +20,7 @@
 from auto_round.utils import convert_dtype_torch2str, logger
 
 try:
-    import auto_round_kernel as ark
+    from auto_round_extension.ark import auto_round_kernel as ark
 
     ARK_INSTALLED = True
 except: