work in progress

Author: smazouz42

pyccel/codegen/compiling/compilers.py

@@ -137,8 +137,8 @@ def _get_exec(self, accelerators):
         os.environ['PATH'] = current_path
 
 
-        errors.report(f"Could not find compiler ({exec_cmd})",
-                severity='fatal')
+        # errors.report(f"Could not find compiler ({exec_cmd})",
+        #         severity='fatal')
 
         return exec_loc
 

pyccel/codegen/printing/cucode.py

@@ -15,12 +15,17 @@
 from pyccel.ast.literals            import Nil
 
 from pyccel.errors.errors           import Errors
+from pyccel.ast.core                import Allocate, Deallocate
+
 
 
 errors = Errors()
 
 __all__ = ["CudaCodePrinter"]
 
+c_imports = {n : Import(n, Module(n, (), ())) for n in
+                ['cuda_ndarrays',]}
+
 class CudaCodePrinter(CCodePrinter):
     """
     Print code in CUDA format.
@@ -133,4 +138,21 @@ def _print_ModuleHeader(self, expr):
                           global_variables,
                           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))
+    
+        alloc_code = f"{self._print(expr.variable)} = cuda_array_create();\n"
+        return f'{alloc_code}'
+        # print(shape)
+        
+        # return "hjsjkahsjkajskasjkasj"
 

pyccel/stdlib/cuda_ndarrays/cuda_ndarrays.cu

@@ -0,0 +1,22 @@
+#include "cuda_ndarrays.h"
+
+void *cuda_array_create(int shape[])
+{
+    size_t i = 0;
+    size_t alloc_size = 1;
+
+    while (shape[i] != 0)
+    {
+        alloc_size *= shape[i];
+        i++;
+    }
+
+    void *array_ptr = malloc(alloc_size);
+    if (array_ptr == NULL)
+    {
+        cout << "Error allocating memory" << endl;
+        return NULL;
+    }
+
+    return array_ptr;
+}

pyccel/stdlib/cuda_ndarrays/cuda_ndarrays.h

@@ -0,0 +1,10 @@
+#ifndef CUDA_NDARRAYS_H
+# define CUDA_NDARRAYS_H
+
+# include <cuda_runtime.h>
+# include <iostream>
+
+using namespace std;
+
+
+#endif