소스 검색

fix previous commit

Cédric Augonnet 15 년 전
부모
커밋
9711586171
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. 5 3
      mpi/starpu_mpi_datatype.c

+ 5 - 3
mpi/starpu_mpi_datatype.c

@@ -30,10 +30,12 @@ typedef void *(*handle_to_ptr_func)(starpu_data_handle);
 
 static int handle_to_datatype_blas(starpu_data_handle data_handle, MPI_Datatype *datatype)
 {
-	unsigned nx = starpu_get_vector_nx(data_handle);
-	size_t elemsize = starpu_get_vector_elemsize(data_handle);
+	unsigned nx = starpu_get_blas_nx(data_handle);
+	unsigned ny = starpu_get_blas_ny(data_handle);
+	unsigned ld = starpu_get_blas_local_ld(data_handle);
+	size_t elemsize = starpu_get_blas_elemsize(data_handle);
 
-	MPI_Type_contiguous(nx*elemsize, MPI_BYTE, datatype);
+	MPI_Type_vector(ny, nx*elemsize, ld*elemsize, MPI_BYTE, datatype);
 	MPI_Type_commit(datatype);
 
 	return 0;