Skip to content

Commit

Permalink
Merge pull request #111 from abouteiller/bugfix/warmup_recursive_skip
Browse files Browse the repository at this point in the history 
When recursive is disabled, first gpu is at index 1
  • Loading branch information
@abouteiller
abouteiller authored Feb 14, 2024
2 parents 155df95 + a8203a3 commit 36d60da
Show file tree
Hide file tree
Showing 9 changed files with 18 additions and 9 deletions.
3 changes: 2 additions & 1 deletion tests/testing_zgemm.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -335,7 +335,8 @@ static void warmup_zgemm(int rank, int nodes, int random_seed, parsec_context_t

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
for(int i = 0; i < MT; i++) {
for(int j = 0; j < NT; j++) {
if( rank == (int)dcC.super.super.rank_of(&dcC.super.super, i, j) ) {
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zgemm_dtd.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -696,7 +696,8 @@ static void warmup_zgemm(int rank, int nodes, int random_seed, parsec_context_t

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
for(int i = 0; i < MT; i++) {
for(int j = 0; j < NT; j++) {
if( rank == (int)dcC.super.super.rank_of(&dcC.super.super, i, j) ) {
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zgeqrf.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -413,7 +413,8 @@ static void warmup_zgeqrf(int rank, int random_seed, parsec_context_t *parsec)
dplasma_zgeqrf_Destruct(zgeqrf);

/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
for(int i = 0; i < MT; i++) {
for(int j = 0; j < NT; j++) {
parsec_data_t *dta = dcA.super.super.data_of(&dcA.super.super, i, j);
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zgeqrf_dtd.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -644,7 +644,8 @@ static void warmup_zgeqrf(int rank, int random_seed, parsec_context_t *parsec)
dplasma_zgeqrf_Destruct(zgeqrf);

/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
for(int i = 0; i < MT; i++) {
for(int j = 0; j < NT; j++) {
parsec_data_t *dta = dcA.super.super.data_of(&dcA.super.super, i, j);
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zgeqrf_dtd_untied.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -697,7 +697,8 @@ static void warmup_zgeqrf(int rank, int random_seed, parsec_context_t *parsec)
dplasma_zgeqrf_Destruct(zgeqrf);

/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
for(int i = 0; i < MT; i++) {
for(int j = 0; j < NT; j++) {
parsec_data_t *dta = dcA.super.super.data_of(&dcA.super.super, i, j);
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zpoinv.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,8 @@ static void warmup_zpoinv(int rank, dplasma_enum_t uplo, int random_seed, parsec

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
if(PARSEC_MATRIX_LOWER == uplo) {
for(int i = 0; i < MT; i++) {
for(int j = 0; j <= i; j++) {
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zpotrf.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -175,7 +175,8 @@ static void warmup_zpotrf(int rank, dplasma_enum_t uplo, int random_seed, parsec

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
if(PARSEC_MATRIX_LOWER == uplo) {
for(int i = 0; i < MT; i++) {
for(int j = 0; j <= i; j++) {
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zpotrf_dtd.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -427,7 +427,8 @@ static void warmup_zpotrf(int rank, dplasma_enum_t uplo, int random_seed, parsec

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
if(PARSEC_MATRIX_LOWER == uplo) {
for(int i = 0; i < MT; i++) {
for(int j = 0; j <= i; j++) {
Expand Down
3 changes: 2 additions & 1 deletion tests/testing_zpotrf_dtd_untied.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -526,7 +526,8 @@ static void warmup_zpotrf(int rank, dplasma_enum_t uplo, int random_seed, parsec

/* Now do the other devices, skipping RECURSIVE */
/* We know that there is a GPU-enabled version of this operation, so warm it up if some device is enabled */
for(did = 2; did < (int)parsec_nb_devices; did++) {
for(did = 1; did < (int)parsec_nb_devices; did++) {
if(!parsec_mca_device_is_gpu(did)) continue;
if(PARSEC_MATRIX_LOWER == uplo) {
for(int i = 0; i < MT; i++) {
for(int j = 0; j <= i; j++) {
Expand Down

0 comments on commit 36d60da

Please sign in to comment.