1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- #ifndef __BLAS_MODEL_H__
- #define __BLAS_MODEL_H__
- #include <starpu.h>
- double gemm_cost(struct starpu_task *task, unsigned nimpl);
- static struct starpu_perfmodel starpu_sgemm_model =
- {
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "sgemm_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "sgemm_goto"
- #else
- .symbol = "sgemm"
- #endif
- };
- static struct starpu_perfmodel starpu_sgemm_model_common =
- {
- .cost_function = gemm_cost,
- .type = STARPU_COMMON,
- };
- static struct starpu_perfmodel starpu_dgemm_model =
- {
- .type = STARPU_HISTORY_BASED,
- #ifdef STARPU_ATLAS
- .symbol = "dgemm_atlas"
- #elif defined(STARPU_GOTO)
- .symbol = "dgemm_goto"
- #else
- .symbol = "dgemm"
- #endif
- };
- static struct starpu_perfmodel starpu_dgemm_model_common =
- {
- .cost_function = gemm_cost,
- .type = STARPU_COMMON,
- };
- #endif
|