work in progress

Author: smazouz42

pyccel/codegen/printing/cucode.py

@@ -139,17 +139,8 @@ def _print_ModuleHeader(self, expr):
                           function_declaration,
                           "#endif // {name.upper()}_H\n"))
     def _print_Allocate(self, expr):
-        self.add_import('cuda_ndarrays')
-        free_code = ''
-        
-        
-        #free the array if its already allocated and checking if its not null if the status is unknown
-        # if  (expr.status == 'unknown'):
-        #     free_code = 'if (%s.shape != NULL)\n' % self._print(expr.variable.name)
-        #     free_code += "{{\n{}}}\n".format(self._print(Deallocate(expr.variable)))
-        # elif  (expr.status == 'allocated'):
-        #     free_code += self._print(Deallocate(expr.variable))
-    
+ 
+        self.add_import(c_imports['cuda_ndarrays'])
         alloc_code = f"{self._print(expr.variable)} = cuda_array_create();\n"
         return f'{alloc_code}'
         # print(shape)

pyccel/codegen/utilities.py

@@ -36,6 +36,7 @@
 # The compile object folder will be in the pyccel dirpath
 internal_libs = {
     "ndarrays"     : ("ndarrays", CompileObj("ndarrays.c",folder="ndarrays")),
+    "cuda_ndarrays": ("cuda_ndarrays", CompileObj("cuda_ndarrays.cu",folder="ndarrays")),
     "pyc_math_f90" : ("math", CompileObj("pyc_math_f90.f90",folder="math")),
     "pyc_math_c"   : ("math", CompileObj("pyc_math_c.c",folder="math")),
     "cwrapper"     : ("cwrapper", CompileObj("cwrapper.c",folder="cwrapper", accelerators=('python',))),