123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288 |
- #include "lu_kernels_model.h"
- #ifdef USE_PERTURBATION
- #define PERTURBATE(a) ((starpu_drand48()*2.0f*(AMPL) + 1.0f - (AMPL))*(a))
- #else
- #define PERTURBATE(a) (a)
- #endif
- double task_11_cost(struct starpu_task *task, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/537.5);
- return PERTURBATE(cost);
- }
- double task_12_cost(struct starpu_task *task, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/3210.80);
-
- return PERTURBATE(cost);
- }
- double task_21_cost(struct starpu_task *task, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/3691.53);
-
- return PERTURBATE(cost);
- }
- double task_22_cost(struct starpu_task *task, unsigned nimpl)
- {
- uint32_t nx, ny, nz;
- nx = starpu_matrix_get_nx(task->handles[2]);
- ny = starpu_matrix_get_ny(task->handles[2]);
- nz = starpu_matrix_get_ny(task->handles[0]);
- double cost = ((nx*ny*nz)/4110.0);
- return PERTURBATE(cost);
- }
- double task_11_cost_cuda(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/1853.7806);
- return PERTURBATE(cost);
- }
- double task_12_cost_cuda(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/42838.5718);
- return PERTURBATE(cost);
- }
- double task_21_cost_cuda(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/49208.667);
- return PERTURBATE(cost);
- }
- double task_22_cost_cuda(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t nx, ny, nz;
- nx = starpu_matrix_get_nx(task->handles[2]);
- ny = starpu_matrix_get_ny(task->handles[2]);
- nz = starpu_matrix_get_ny(task->handles[0]);
- double cost = ((nx*ny*nz)/57523.560);
- return PERTURBATE(cost);
- }
- double task_11_cost_cpu(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/537.5);
- return PERTURBATE(cost);
- }
- double task_12_cost_cpu(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/6668.224);
- return PERTURBATE(cost);
- }
- double task_21_cost_cpu(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t n;
- n = starpu_matrix_get_nx(task->handles[0]);
- double cost = ((n*n*n)/6793.8423);
- return PERTURBATE(cost);
- }
- double task_22_cost_cpu(struct starpu_task *task, enum starpu_perf_archtype arch, unsigned nimpl)
- {
- uint32_t nx, ny, nz;
- nx = starpu_matrix_get_nx(task->handles[2]);
- ny = starpu_matrix_get_ny(task->handles[2]);
- nz = starpu_matrix_get_ny(task->handles[0]);
- double cost = ((nx*ny*nz)/4203.0175);
- return PERTURBATE(cost);
- }
- struct starpu_perfmodel model_11 =
- {
- .cost_function = task_11_cost,
- .per_arch =
- {
- [STARPU_CPU_DEFAULT][0] = { .cost_function = task_11_cost_cpu },
- [STARPU_CUDA_DEFAULT][0] = { .cost_function = task_11_cost_cuda }
- },
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "lu_model_11_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "lu_model_11_goto"
- #else
- .symbol = "lu_model_11"
- #endif
- };
- struct starpu_perfmodel model_12 =
- {
- .cost_function = task_12_cost,
- .per_arch =
- {
- [STARPU_CPU_DEFAULT][0] = { .cost_function = task_12_cost_cpu },
- [STARPU_CUDA_DEFAULT][0] = { .cost_function = task_12_cost_cuda }
- },
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "lu_model_12_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "lu_model_12_goto"
- #else
- .symbol = "lu_model_12"
- #endif
- };
- struct starpu_perfmodel model_21 =
- {
- .cost_function = task_21_cost,
- .per_arch =
- {
- [STARPU_CPU_DEFAULT][0] = { .cost_function = task_21_cost_cpu },
- [STARPU_CUDA_DEFAULT][0] = { .cost_function = task_21_cost_cuda }
- },
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "lu_model_21_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "lu_model_21_goto"
- #else
- .symbol = "lu_model_21"
- #endif
- };
- struct starpu_perfmodel model_22 =
- {
- .cost_function = task_22_cost,
- .per_arch =
- {
- [STARPU_CPU_DEFAULT][0] = { .cost_function = task_22_cost_cpu },
- [STARPU_CUDA_DEFAULT][0] = { .cost_function = task_22_cost_cuda }
- },
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "lu_model_22_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "lu_model_22_goto"
- #else
- .symbol = "lu_model_22"
- #endif
- };
|