starpu_py_np.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. # StarPU --- Runtime system for heterogeneous multicore architectures.
  2. #
  3. # Copyright (C) 2020 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
  4. #
  5. # StarPU is free software; you can redistribute it and/or modify
  6. # it under the terms of the GNU Lesser General Public License as published by
  7. # the Free Software Foundation; either version 2.1 of the License, or (at
  8. # your option) any later version.
  9. #
  10. # StarPU is distributed in the hope that it will be useful, but
  11. # WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. #
  14. # See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. #
  16. import starpu
  17. import asyncio
  18. import numpy as np
  19. ###############################################################################
  20. def scal(a, t):
  21. for i in range(len(t)):
  22. t[i]=t[i]*a
  23. return t
  24. t=np.array([1,2,3,4,5,6,7,8,9,10])
  25. async def main():
  26. fut8 = starpu.task_submit()(scal, 2, t)
  27. res8 = await fut8
  28. print("The result of Example 10 is", res8)
  29. print("The return array is", t)
  30. #print("The result type is", type(res8))
  31. asyncio.run(main())
  32. #starpu.task_wait_for_all()