blas_model.h 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. /*
  2. * StarPU
  3. * Copyright (C) Université Bordeaux 1, CNRS 2008-2010 (see AUTHORS file)
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #ifndef __BLAS_MODEL_H__
  17. #define __BLAS_MODEL_H__
  18. #include <starpu.h>
  19. double gemm_cost(starpu_buffer_descr *descr);
  20. static struct starpu_perfmodel_t sgemm_model = {
  21. .cost_model = gemm_cost,
  22. .type = STARPU_HISTORY_BASED,
  23. #ifdef STARPU_ATLAS
  24. .symbol = "sgemm_atlas"
  25. #elif defined(STARPU_GOTO)
  26. .symbol = "sgemm_goto"
  27. #else
  28. .symbol = "sgemm"
  29. #endif
  30. };
  31. static struct starpu_perfmodel_t sgemm_model_common = {
  32. .cost_model = gemm_cost,
  33. .type = STARPU_COMMON,
  34. };
  35. #endif // __BLAS_MODEL_H__