mult_generated_graph.jl 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  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. if length(ARGS) != 6
  17. println("Usage : julia prog.jl start_dim step_dim stop_dim nb_tests nslicesx nslicesy")
  18. quit()
  19. end
  20. include("../../src/Wrapper/Julia/starpu_include.jl")
  21. using StarPU
  22. @debugprint "starpu_init"
  23. starpu_init(extern_task_path = "../build/generated_tasks.so")
  24. perfmodel = StarpuPerfmodel(
  25. perf_type = STARPU_HISTORY_BASED,
  26. symbol = "history_perf"
  27. )
  28. cl = StarpuCodelet(
  29. cpu_func = "matrix_mult",
  30. gpu_func = "CUDA_matrix_mult",
  31. modes = [STARPU_R, STARPU_R, STARPU_W],
  32. perfmodel = perfmodel
  33. )
  34. include("mult_def.jl")
  35. display_times(map( (x -> parse(Int64,x)) , ARGS)..., "../mult_generated.dat")
  36. @debugprint "starpu_shutdown"
  37. starpu_shutdown()