unimore/hpc-2022-g3
unimore/hpc-2022-g3
1
Fork 0
mirror of https://github.com/Steffo99/unimore-hpc-assignments.git synced 2024-11-22 08:04:25 +00:00
Code Activity

Gate CUDA usage behind the POLYBENCH_USE_CUDA flag

Browse source
This commit is contained in:
Steffo 2022-11-29 17:47:34 +01:00 committed by GitHub
parent f7cbdba3a2
commit b526108132
Signed by: github
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
atax/atax.cu
View file

@ -108,33 +108,41 @@ __host__ static void kernel_atax(DATA_TYPE** A, DATA_TYPE* X, DATA_TYPE* Y)
*/
__host__ int main(int argc, char** argv)
{
// A[NX][NY]
DATA_TYPE** A = new DATA_TYPE*[NX] {};
for(unsigned int x = 0; x < NX; x++)
{
A[x] = new DATA_TYPE[NY] {};
}
#ifndef POLYBENCH_USE_CUDA
// A[NX][NY]
DATA_TYPE** A = new DATA_TYPE*[NX] {};
for(unsigned int x = 0; x < NX; x++)
{
A[x] = new DATA_TYPE[NY] {};
}
DATA_TYPE* x = new DATA_TYPE[NY] {};
DATA_TYPE* y = new DATA_TYPE[NX] {};
#ifdef POLYBENCH_INCLUDE_INIT
polybench_start_instruments;
#endif
init_array(A, x, y);
#ifndef POLYBENCH_INCLUDE_INIT
polybench_start_instruments;
#endif
kernel_atax(A, x, y);
polybench_stop_instruments;
polybench_print_instruments;
/* Prevent dead-code elimination. All live-out data must be printed by the function call in argument. */
polybench_prevent_dce(print_array(y));
#else
DATA_TYPE* x = new DATA_TYPE[NY] {};
DATA_TYPE* y = new DATA_TYPE[NX] {};
#ifdef POLYBENCH_INCLUDE_INIT
polybench_start_instruments;
#endif
init_array(A, x, y);
#ifndef POLYBENCH_INCLUDE_INIT
polybench_start_instruments;
#endif
kernel_atax(A, x, y);
polybench_stop_instruments;
polybench_print_instruments;
/* Prevent dead-code elimination. All live-out data must be printed by the function call in argument. */
polybench_prevent_dce(print_array(y));
return 0;
}