[WIP] Task

pyccel/ast/core.py

@@ -3821,6 +3821,64 @@ def __reduce_ex__(self, i):
         mod_name  = self.mod_name)
         return (apply, (self.__class__, (), kwargs))
 
+
+class TaskFunctionCall(FunctionCall):
+    pass
+
+class Task(Basic):
+    """
+    """
+    __slots__ = ('_stmt','_inputs','_outputs','_preceders', '_shareds', '_should_wait')
+    _attribute_nodes = ('_stmt','_inputs','_preceders')
+
+    def __init__(self, stmt, inputs = (), outputs = dict(), shareds = set(), preceders = ()):
+        self._stmt        = stmt
+        self._inputs      = inputs
+        self._outputs     = outputs
+        self._preceders   = preceders
+        self._shareds     = shareds
+        self._should_wait = False
+        super().__init__()
+
+    @property
+    def stmt(self):
+        return self._stmt
+
+    @property
+    def inputs(self):
+        return self._inputs
+
+    @property
+    def outputs(self):
+        return self._outputs
+
+    @property
+    def preceders(self):
+        return self._preceders
+
+    @property
+    def shareds(self):
+        return self._shareds
+
+    @property
+    def should_wait(self):
+        return self._should_wait
+
+    @should_wait.setter
+    def should_wait(self, value):
+        self._should_wait = value
+
+class TaskMaster(Task):
+    __slots__ = ('_num_threads',)
+
+    def __init__(self, num_threads, *args, **kwargs):
+        self._num_threads = num_threads
+        super().__init__(*args, **kwargs)
+
+    @property
+    def num_threads(self):
+        return self._num_threads
+
 #==============================================================================
 def process_shape(shape):
     if not hasattr(shape,'__iter__'):

pyccel/ast/sympy_helper.py

@@ -11,7 +11,7 @@
 import sympy as sp
 from sympy.core.numbers import One, NegativeOne, Zero, Half
 
-from .operators import PyccelAdd, PyccelMul, PyccelPow, PyccelUnarySub
+from .operators import PyccelAdd, PyccelMul, PyccelPow, PyccelUnarySub, PyccelFloorDiv
 from .operators import PyccelDiv, PyccelMinus, PyccelAssociativeParenthesis
 from .core      import create_incremented_string
 
@@ -169,6 +169,10 @@ def pyccel_to_sympy(expr, symbol_map, used_names):
     elif isinstance(expr, MathCeil):
         return sp.ceiling(pyccel_to_sympy(expr.args[0], symbol_map, used_names))
 
+    elif isinstance(expr, PyccelFloorDiv):
+        args = [pyccel_to_sympy(e, symbol_map, used_names) for e in expr.args]
+        return args[0] // args[1]
+
     elif expr in symbol_map.values():
         return list(symbol_map.keys())[list(symbol_map.values()).index(expr)]
 

pyccel/codegen/pipeline.py

@@ -328,7 +328,10 @@ def handle_error(stage):
                     lib_dest_path = os.path.join(pyccel_dirpath, lib_name)
                     with FileLock(lib_dest_path + '.lock'):
                         if not os.path.exists(lib_dest_path):
-                            shutil.copytree(lib_path, lib_dest_path)
+                            try:
+                                shutil.copytree(lib_path, lib_dest_path)
+                            except:
+                                print('[Warning] : can\'t copy file permissions')
 
                     # stop after copying lib to __pyccel__ directory for
                     # convert only

pyccel/codegen/printing/ccode.py

@@ -1600,9 +1600,36 @@ def _print_CommentBlock(self, expr):
     def _print_EmptyNode(self, expr):
         return ''
 
+    def _print_Task(self, expr):
+        inouts = []
+        inputs = expr.inputs
+
+        # resolve duplicated variables
+        outputs = [a if a not in inputs else inouts.append(a) for a in expr.outputs.keys()] if expr.outputs.keys() else None
+        inputs = [a for a in inputs if a not in inouts] if inputs else None
+
+        #printing
+        outputs = 'out:{}'.format(','.join(self._print(a) for a in outputs)) if outputs else ''
+        inputs = 'in:{}'.format(','.join(self._print(a) for a in inputs)) if inputs else ''
+        inouts = 'inout:{}'.format(','.join(self._print(a) for a in inouts)) if inouts else ''
+
+        depend = 'depend {}'.format(outputs + inputs + inouts) if inputs or outputs or inouts else ''
+
+        should_wait = '#pragma omp taskwait' if expr.should_wait else ''
+
+        start_task = '#pragma omp task {}'.format(depend) + '{\n'
+        structured_code_block = self._print(expr.stmt)
+        end_task = "}\n"
+
+        code = should_wait + start_task + structured_code_block + end_task
+        return code
+
+
+
     #=================== OMP ==================
 
     def _print_OmpAnnotatedComment(self, expr):
+        print(expr)
         clauses = ''
         if expr.combined:
             clauses = ' ' + expr.combined

pyccel/codegen/printing/fcode.py

@@ -1851,6 +1851,56 @@ def _do_range(target, iterable, prolog, epilog):
                 '{body}'
                 '{epilog}').format(prolog=prolog, body=body, epilog=epilog)
 
+    def _print_Task(self, expr):
+
+        inputs  = expr.inputs
+        outputs = tuple(i for i, v in expr.outputs.items() if v is True)
+        shareds = expr.shareds
+
+        inputs  = ' depend(in:{})'.format(','.join(a.name for a in inputs)) if inputs else ''
+        outputs = ' depend(out:{})'.format(','.join(a.name for a in outputs)) if outputs else ''
+        shareds = ' shared({})'.format(', '.join(a.name for a in shareds)) if shareds else ''
+
+        code = ('!$omp task{shareds}{inputs}{outputs}\n'
+               '{statement}'
+               '!$omp end task\n'
+               '{should_wait}\n').format(shareds = shareds,
+                                       inputs = inputs,
+                                       outputs = outputs,
+                                       statement = self._print(expr.stmt),
+                                       should_wait = '!$omp taskwait' if expr.should_wait else '')
+        return code
+
+    def _print_TaskMaster(self, expr):
+        threads_num = ' num_threads({})'.format(self._print(expr.num_threads.python_value)) if expr.num_threads else ''
+
+        code = ('!$omp parallel{threads_num}\n'
+                '!$omp single\n'
+                '{statement}'
+                '!$omp end single\n'
+                '!$omp end parallel\n').format(threads_num = threads_num, statement = self._print(expr.stmt))
+
+        return code
+        # resolve duplicated variables
+        #outputs = [a if a not in inputs else inouts.append(a) for a in expr.outputs.keys()] if expr.outputs.keys() else None
+        #inputs = [a for a in inputs if a not in inouts] if inputs else None
+
+        #printing
+        #outputs = 'out:{}'.format(','.join(self._print(a) for a in outputs)) if outputs else ''
+        #inputs = 'in:{}'.format(','.join(self._print(a) for a in inputs)) if inputs else ''
+        #inouts = 'inout:{}'.format(','.join(self._print(a) for a in inouts)) if inouts else ''
+
+        #depend = 'depend {}'.format(outputs + inputs + inouts) if inputs or outputs or inouts else ''
+        #should_wait = '!$omp taskwait\n' if expr.should_wait else ''
+
+        #start_task = "!$omp task {}\n".format(depend)
+        #structured_code_block = self._print(expr.stmt)
+        #end_task = "!$omp end task\n"
+
+
+        #code = '{}'.format(start_task + structured_code_block + end_task + should_wait)
+        #return code
+
     # .....................................................
     #               Print OpenMP AnnotatedComment
     # .....................................................

pyccel/decorators.py

@@ -20,7 +20,8 @@
     'private',
     'elemental',
     'stack_array',
-    'allow_negative_index'
+    'allow_negative_index',
+    'task'
 )
 
 def lambdify(f):
@@ -96,3 +97,20 @@ def allow_negative_index(f,*args):
     def identity(f):
         return f
     return identity
+
+
+def task(type = 'child', threads_num = None):
+    """
+    Decorator indicate that the function can be used in multithreading
+
+    Prameters
+    ---------
+    type : string
+        indecate that the current function is responsible of running other
+        task functions ('master') or a task function ('child' / None)
+    threads_num : integer
+        number of threads reserved for this parallel region
+    """
+    def identity(f):
+        return f
+    return identity

pyccel/parser/base.py

@@ -150,6 +150,8 @@ def __init__(self, *, headers=None, decorators=None, templates=None):
         self._symbolic_functions = OrderedDict()
         self._python_functions   = OrderedDict()
 
+        self._tasks              = []
+
         self._is_loop = False
         # scoping for loops
         self._loops = []
@@ -246,6 +248,11 @@ def is_loop(self):
     def loops(self):
         return self._loops
 
+    @property
+    def tasks(self):
+        """
+        """
+        return self._tasks