mult_generated.jl 929 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  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. clcpu = StarpuCodelet(
  20. cpu_func = "matrix_mult",
  21. modes = [STARPU_R, STARPU_R, STARPU_W],
  22. perfmodel = perfmodel
  23. )
  24. clgpu = StarpuCodelet(
  25. gpu_func = "CUDA_matrix_mult",
  26. modes = [STARPU_R, STARPU_R, STARPU_W],
  27. perfmodel = perfmodel
  28. )
  29. include("mult_def.jl")
  30. display_times(map( (x -> parse(Int64,x)) , ARGS)...)
  31. @debugprint "starpu_shutdown"
  32. starpu_shutdown()