Skip to content

Commit

Permalink
So far only NN gemm can run with HIP
Browse files Browse the repository at this point in the history
Signed-off-by: Aurelien Bouteiller <[email protected]>
  • Loading branch information
abouteiller committed Jun 5, 2023
1 parent 6022fad commit fae375c
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions src/zgemm_wrapper.c
Original file line number Diff line number Diff line change
Expand Up @@ -91,12 +91,14 @@ dplasma_zgemm_summa_new(dplasma_enum_t transA, dplasma_enum_t transB,
parsec_zgemm_NT_summa_taskpool_t* tp;
tp = parsec_zgemm_NT_summa_new(transA, transB, alpha, beta,
ddc_A, ddc_B, ddc_C, (parsec_data_collection_t*)Cdist);
#if 0
#if defined(PARSEC_HAVE_HIP)
/* It doesn't cost anything to define these infos if we have HIP but
* don't have GPUs on the current machine, so we do it non-conditionally */
tp->_g_hip_handles_infokey = parsec_info_lookup(&parsec_per_stream_infos, "DPLASMA::HIP::HANDLES", NULL);
#else
tp->_g_hip_handles_infokey = PARSEC_INFO_ID_UNDEFINED;
#endif
#endif
zgemm_tp = (parsec_taskpool_t*)tp;
}
Expand All @@ -106,12 +108,14 @@ dplasma_zgemm_summa_new(dplasma_enum_t transA, dplasma_enum_t transB,
parsec_zgemm_TN_summa_taskpool_t* tp;
tp = parsec_zgemm_TN_summa_new(transA, transB, alpha, beta,
ddc_A, ddc_B, ddc_C, (parsec_data_collection_t*)Cdist);
#if 0
#if defined(PARSEC_HAVE_HIP)
/* It doesn't cost anything to define these infos if we have HIP but
* don't have GPUs on the current machine, so we do it non-conditionally */
tp->_g_hip_handles_infokey = parsec_info_lookup(&parsec_per_stream_infos, "DPLASMA::HIP::HANDLES", NULL);
#else
tp->_g_hip_handles_infokey = PARSEC_INFO_ID_UNDEFINED;
#endif
#endif
zgemm_tp = (parsec_taskpool_t*)tp;
} else {
Expand All @@ -120,12 +124,14 @@ dplasma_zgemm_summa_new(dplasma_enum_t transA, dplasma_enum_t transB,
tp = parsec_zgemm_TT_summa_new(transA, transB, alpha, beta,
ddc_A, ddc_B, ddc_C,
(parsec_data_collection_t*)Cdist);
#if 0
#if defined(PARSEC_HAVE_HIP)
/* It doesn't cost anything to define these infos if we have HIP but
* don't have GPUs on the current machine, so we do it non-conditionally */
tp->_g_hip_handles_infokey = parsec_info_lookup(&parsec_per_stream_infos, "DPLASMA::HIP::HANDLES", NULL);
#else
tp->_g_hip_handles_infokey = PARSEC_INFO_ID_UNDEFINED;
#endif
#endif
zgemm_tp = (parsec_taskpool_t*)tp;
}
Expand Down

0 comments on commit fae375c

Please sign in to comment.