Home Esplora Aiuto
Accedi
exa2pro
/
starpu-max
Segui 19
Vota 0
Forka 0
File Problemi 0 Pull Requests 0 Wiki
Albero (Tree): 31f6bb8b27
Rami (Branch) Tag
starpu-max
/
examples
/
cg
/
cg.h

cg.h 2.1 KB
Cronologia Originale

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
    2. 

  2.  *
    3. 

  3.  * Copyright (C) 2010-2011  Université de Bordeaux 1
    4. 

  4.  *
    5. 

  5.  * StarPU is free software; you can redistribute it and/or modify
    6. 

  6.  * it under the terms of the GNU Lesser General Public License as published by
    7. 

  7.  * the Free Software Foundation; either version 2.1 of the License, or (at
    8. 

  8.  * your option) any later version.
    9. 

  9.  *
    10. 

  10.  * StarPU is distributed in the hope that it will be useful, but
    11. 

  11.  * WITHOUT ANY WARRANTY; without even the implied warranty of
    12. 

  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
    13. 

  13.  *
    14. 

  14.  * See the GNU Lesser General Public License in COPYING.LGPL for more details.
    15. 

  15.  */
    16. 

  16. 

  17. #ifndef __STARPU_EXAMPLE_CG_H__
    18. 

  18. #define __STARPU_EXAMPLE_CG_H__
    19. 

  19. 

  20. #include <starpu.h>
    21. 

  21. #include <math.h>
    22. 

  22. #include <common/blas.h>
    23. 

  23. 

  24. #ifdef STARPU_USE_CUDA
    25. 

  25. #include <cuda.h>
    26. 

  26. #include <cublas.h>
    27. 

  27. #endif
    28. 

  28. 

  29. #define DOUBLE
    30. 

  30. 

  31. #ifdef DOUBLE
    32. 

  32. #define TYPE	double
    33. 

  33. #define GEMV	DGEMV
    34. 

  34. #define DOT	DDOT
    35. 

  35. #define GEMV	DGEMV
    36. 

  36. #define AXPY	DAXPY
    37. 

  37. #define SCAL	DSCAL
    38. 

  38. #define cublasdot	cublasDdot
    39. 

  39. #define cublasscal	cublasDscal
    40. 

  40. #define cublasaxpy	cublasDaxpy
    41. 

  41. #define cublasgemv	cublasDgemv
    42. 

  42. #define cublasscal	cublasDscal
    43. 

  43. #else
    44. 

  44. #define TYPE	float
    45. 

  45. #define GEMV	SGEMV
    46. 

  46. #define DOT	SDOT
    47. 

  47. #define GEMV	SGEMV
    48. 

  48. #define AXPY	SAXPY
    49. 

  49. #define SCAL	SSCAL
    50. 

  50. #define cublasdot	cublasSdot
    51. 

  51. #define cublasscal	cublasSscal
    52. 

  52. #define cublasaxpy	cublasSaxpy
    53. 

  53. #define cublasgemv	cublasSgemv
    54. 

  54. #define cublasscal	cublasSscal
    55. 

  55. #endif
    56. 

  56. 

  57. int dot_kernel(starpu_data_handle_t v1,
    58. 

  58. 	       starpu_data_handle_t v2,
    59. 

  59. 	       starpu_data_handle_t s,
    60. 

  60. 	       unsigned nblocks,
    61. 

  61. 	       int use_reduction);
    62. 

  62. 

  63. int gemv_kernel(starpu_data_handle_t v1,
    64. 

  64.                 starpu_data_handle_t matrix, 
    65. 

  65.                 starpu_data_handle_t v2,
    66. 

  66.                 TYPE p1, TYPE p2,
    67. 

  67. 		unsigned nblocks,
    68. 

  68. 		int use_reduction);
    69. 

  69. 

  70. int axpy_kernel(starpu_data_handle_t v1,
    71. 

  71. 		starpu_data_handle_t v2, TYPE p1,
    72. 

  72. 		unsigned nblocks);
    73. 

  73. 

  74. int scal_axpy_kernel(starpu_data_handle_t v1, TYPE p1,
    75. 

  75. 		     starpu_data_handle_t v2, TYPE p2,
    76. 

  76. 		     unsigned nblocks);
    77. 

  77. 

  78. int copy_handle(starpu_data_handle_t dst,
    79. 

  79. 		starpu_data_handle_t src,
    80. 

  80. 		unsigned nblocks);
    81. 

  81. 

  82. #endif /* __STARPU_EXAMPLE_CG_H__ */
    83. 

  83.