Commit 4afab39d authored by Mathieu Faverge's avatar Mathieu Faverge

Fix axpy interface

parent 0f2ad71b
...@@ -34,6 +34,10 @@ void INSERT_TASK_dzasum( const RUNTIME_option_t *options, ...@@ -34,6 +34,10 @@ void INSERT_TASK_dzasum( const RUNTIME_option_t *options,
cham_store_t storev, cham_uplo_t uplo, int M, int N, cham_store_t storev, cham_uplo_t uplo, int M, int N,
const CHAM_desc_t *A, int Am, int An, int lda, const CHAM_desc_t *A, int Am, int An, int lda,
const CHAM_desc_t *B, int Bm, int Bn ); const CHAM_desc_t *B, int Bm, int Bn );
void INSERT_TASK_zaxpy( const RUNTIME_option_t *options,
int M, CHAMELEON_Complex64_t alpha,
const CHAM_desc_t *A, int Am, int An, int incA,
const CHAM_desc_t *B, int Bm, int Bn, int incB );
void INSERT_TASK_zgeadd( const RUNTIME_option_t *options, void INSERT_TASK_zgeadd( const RUNTIME_option_t *options,
cham_trans_t trans, int m, int n, int nb, cham_trans_t trans, int m, int n, int nb,
CHAMELEON_Complex64_t alpha, const CHAM_desc_t *A, int Am, int An, int lda, CHAMELEON_Complex64_t alpha, const CHAM_desc_t *A, int Am, int An, int lda,
......
...@@ -49,11 +49,11 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options, ...@@ -49,11 +49,11 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options,
parsec_dtd_taskpool_insert_task( parsec_dtd_taskpool_insert_task(
PARSEC_dtd_taskpool, CORE_zaxpy_parsec, options->priority, "axpy", PARSEC_dtd_taskpool, CORE_zaxpy_parsec, options->priority, "axpy",
sizeof(int), &M, VALUE, sizeof(int), &M, VALUE,
sizeof(CHAMELEON_Complex64_t), &alpha, VALUE, sizeof(CHAMELEON_Complex64_t), &alpha, VALUE,
PASSED_BY_REF, RTBLKADDR( A, CHAMELEON_Complex64_t, Am, An ), chameleon_parsec_get_arena_index( A ) | INPUT, PASSED_BY_REF, RTBLKADDR( A, CHAMELEON_Complex64_t, Am, An ), chameleon_parsec_get_arena_index( A ) | INPUT,
sizeof(int), &incA, VALUE, sizeof(int), &incA, VALUE,
PASSED_BY_REF, RTBLKADDR( B, CHAMELEON_Complex64_t, Bm, Bn ), chameleon_parsec_get_arena_index( B ) | INOUT | AFFINITY, PASSED_BY_REF, RTBLKADDR( B, CHAMELEON_Complex64_t, Bm, Bn ), chameleon_parsec_get_arena_index( B ) | INOUT | AFFINITY,
sizeof(int), &incB, VALUE, sizeof(int), &incB, VALUE,
PARSEC_DTD_ARG_END ); PARSEC_DTD_ARG_END );
} }
...@@ -37,7 +37,7 @@ void CORE_zaxpy_quark(Quark *quark) ...@@ -37,7 +37,7 @@ void CORE_zaxpy_quark(Quark *quark)
} }
void INSERT_TASK_zaxpy(const RUNTIME_option_t *options, void INSERT_TASK_zaxpy(const RUNTIME_option_t *options,
int M, CHAMELEON_Complex64_t *alpha, int M, CHAMELEON_Complex64_t alpha,
const CHAM_desc_t *A, int Am, int An, int incA, const CHAM_desc_t *A, int Am, int An, int incA,
const CHAM_desc_t *B, int Bm, int Bn, int incB) const CHAM_desc_t *B, int Bm, int Bn, int incB)
{ {
...@@ -45,10 +45,10 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options, ...@@ -45,10 +45,10 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options,
DAG_CORE_AXPY; DAG_CORE_AXPY;
QUARK_Insert_Task(opt->quark, CORE_zaxpy_quark, (Quark_Task_Flags*)opt, QUARK_Insert_Task(opt->quark, CORE_zaxpy_quark, (Quark_Task_Flags*)opt,
sizeof(int), &M, VALUE, sizeof(int), &M, VALUE,
sizeof(CHAMELEON_Complex64_t), alpha, VALUE, sizeof(CHAMELEON_Complex64_t), &alpha, VALUE,
sizeof(CHAMELEON_Complex64_t)*M, RTBLKADDR(A, CHAMELEON_Complex64_t, Am, An), INPUT, sizeof(CHAMELEON_Complex64_t)*M, RTBLKADDR(A, CHAMELEON_Complex64_t, Am, An), INPUT,
sizeof(int), &incA, VALUE, sizeof(int), &incA, VALUE,
sizeof(CHAMELEON_Complex64_t)*M, RTBLKADDR(B, CHAMELEON_Complex64_t, Bm, Bn), INOUT, sizeof(CHAMELEON_Complex64_t)*M, RTBLKADDR(B, CHAMELEON_Complex64_t, Bm, Bn), INOUT,
sizeof(int), &incB, VALUE, sizeof(int), &incB, VALUE,
0); 0);
} }
...@@ -36,7 +36,7 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options, ...@@ -36,7 +36,7 @@ void INSERT_TASK_zaxpy(const RUNTIME_option_t *options,
starpu_insert_task( starpu_insert_task(
starpu_mpi_codelet(codelet), starpu_mpi_codelet(codelet),
STARPU_VALUE, &M, sizeof(int), STARPU_VALUE, &M, sizeof(int),
STARPU_VALUE, alpha, sizeof(CHAMELEON_Complex64_t), STARPU_VALUE, &alpha, sizeof(CHAMELEON_Complex64_t),
STARPU_R, RTBLKADDR(A, CHAMELEON_Complex64_t, Am, An), STARPU_R, RTBLKADDR(A, CHAMELEON_Complex64_t, Am, An),
STARPU_VALUE, &incA, sizeof(int), STARPU_VALUE, &incA, sizeof(int),
STARPU_RW, RTBLKADDR(B, CHAMELEON_Complex64_t, Bm, Bn), STARPU_RW, RTBLKADDR(B, CHAMELEON_Complex64_t, Bm, Bn),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment