TREX/irpjast
1
0
Fork 0
mirror of https://github.com/TREX-CoE/irpjast.git synced 2024-10-02 06:21:07 +02:00
Code Issues Projects Releases Wiki Activity

Fixed bug in data_register

Browse Source
This commit is contained in:
Anthony Scemama 2021-04-27 01:18:48 +02:00
parent 569893eb25
commit 382e717d62
2 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Makefile
View File

@ -1,7 +1,9 @@
IRPF90 = irpf90/bin/irpf90 --codelet=factor_een:2 --align=4096 # -s nelec_8:504 -s nnuc:100 -s ncord:5 #-a -d
STARPU=~/StarPU/starpu-1.3.7/starpu-1.3.pc
IRPF90 = irpf90/bin/irpf90 --codelet=factor_een:2 --align=64 # -s nelec_8:504 -s nnuc:100 -s ncord:5 #-a -d
#FC = ifort -xCORE-AVX512 -g -mkl=sequential -qopt-zmm-usage=high
FC = ifort -xCORE-AVX2 -g
CC = gcc -fopenmp $(shell pkg-config --cflags starpu-1.3)
FC = ifort -xCORE-AVX512 -g -mkl=sequential -qopt-zmm-usage=high -traceback
FC = ifort -axCORE-AVX512 -g -mkl=sequential -qopt-zmm-usage=high -traceback
CC = icc -mkl=sequential $(shell pkg-config --cflags $(STARPU)) -C -traceback
FCFLAGS= -O3 -I .
NINJA = ninja
ARCHIVE = ar crs
@ -9,7 +11,7 @@ RANLIB = ranlib
SRC= qmckl_blas_f.f90 qmckl_dgemm.c
OBJ= IRPF90_temp/qmckl_blas_f.o IRPF90_temp/qmckl_dgemm.o
LIB= -mkl=sequential -lgomp $(shell pkg-config --libs starpu-1.3)
LIB= -mkl=sequential $(shell pkg-config --libs $(STARPU))
-include irpf90.make
export

9
qmckl_dgemm.c
View File

@ -2,7 +2,7 @@
#include <starpu.h>
#include <cblas.h>
#include <mkl_cblas.h>
#include <stdint.h>
#include <assert.h>
#include <stdlib.h>
@ -177,6 +177,7 @@ void qmckl_tasks_run(struct dgemm_args** gemms, int ngemms)
int rc = starpu_init(NULL);
assert (rc == 0);
starpu_data_handle_t matrix_handle[ngemms][3];
for (int i=0 ; i<ngemms ; ++i)
{
@ -184,24 +185,24 @@ void qmckl_tasks_run(struct dgemm_args** gemms, int ngemms)
STARPU_MAIN_RAM,
(uintptr_t) gemms[i]->A,
gemms[i]->lda,
gemms[i]->k,
gemms[i]->m,
gemms[i]->k,
sizeof(double));
starpu_matrix_data_register(&(matrix_handle[i][1]),
STARPU_MAIN_RAM,
(uintptr_t) gemms[i]->B,
gemms[i]->ldb,
gemms[i]->n,
gemms[i]->k,
gemms[i]->n,
sizeof(double));
starpu_matrix_data_register(&(matrix_handle[i][2]),
STARPU_MAIN_RAM,
(uintptr_t) gemms[i]->C,
gemms[i]->ldc,
gemms[i]->n,
gemms[i]->m,
gemms[i]->n,
sizeof(double));
struct starpu_task *task = starpu_task_create();