12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- #ifndef __BLAS_MODEL_H__
- #define __BLAS_MODEL_H__
- #include <starpu.h>
- double gemm_cost(starpu_buffer_descr *descr);
- static struct starpu_perfmodel_t 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_t starpu_sgemm_model_common = {
- .cost_model = gemm_cost,
- .type = STARPU_COMMON,
- };
- static struct starpu_perfmodel_t 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_t starpu_dgemm_model_common = {
- .cost_model = gemm_cost,
- .type = STARPU_COMMON,
- };
- #endif
|