From fb95bb1c4e1ad962090ba7ae125e7498681e4e60 Mon Sep 17 00:00:00 2001
From: Vincent TEMPLIER <vincent.templier@cea.fr>
Date: Wed, 26 Jul 2023 07:55:41 +0000
Subject: [PATCH] Add CPU library in aidge package

---
 setup.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/setup.py b/setup.py
index 26ee4410..59151ba6 100644
--- a/setup.py
+++ b/setup.py
@@ -83,14 +83,14 @@ class CMakeBuild(build_ext):
                     shutil.copy(currentFile, str(aidge_package.absolute())) 
 
         # Get "aidge cpu" package
-        # ext_lib = build_temp / "_CPU"
-
-        # # Copy all shared object files from build_temp/lib to build_lib
-        # for root, _, files in os.walk(ext_lib.absolute()):
-        #     for file in files:
-        #         if file.endswith('.so'):
-        #             currentFile=os.path.join(root, file)
-        #             shutil.copy(currentFile, str(aidge_package.absolute()))     
+        ext_lib = build_temp / "aidge" / "_CPU"
+
+        # Copy all shared object files from build_temp/lib to build_lib
+        for root, _, files in os.walk(ext_lib.absolute()):
+            for file in files:
+                if file.endswith('.so'):
+                    currentFile=os.path.join(root, file)
+                    shutil.copy(currentFile, str(aidge_package.absolute()))     
 
 
 if __name__ == '__main__':
-- 
GitLab