mult_generated_graph.jl 697 B

1234567891011121314151617181920212223242526272829303132333435
  1. if length(ARGS) != 6
  2. println("Usage : julia prog.jl start_dim step_dim stop_dim nb_tests nslicesx nslicesy")
  3. quit()
  4. end
  5. include("../../src/Wrapper/Julia/starpu_include.jl")
  6. using StarPU
  7. @debugprint "starpu_init"
  8. starpu_init(extern_task_path = "../build/generated_tasks.so")
  9. perfmodel = StarpuPerfmodel(
  10. perf_type = STARPU_HISTORY_BASED,
  11. symbol = "history_perf"
  12. )
  13. cl = StarpuCodelet(
  14. cpu_func = "matrix_mult",
  15. gpu_func = "CUDA_matrix_mult",
  16. modes = [STARPU_R, STARPU_R, STARPU_W],
  17. perfmodel = perfmodel
  18. )
  19. include("mult_def.jl")
  20. display_times(map( (x -> parse(Int64,x)) , ARGS)..., "../mult_generated.dat")
  21. @debugprint "starpu_shutdown"
  22. starpu_shutdown()