1 static char help[] = "Test MatMatMult() and MatTransposeMatMult() for MPIAIJ and MPIDENSE matrices. \n\ 2 Sequential part of mpidense matrix allows changes made by MatSeqDenseSetLDA(). \n\n"; 3 4 #include <petsc.h> 5 6 int main(int argc, char ** argv) 7 { 8 Mat A, B, C, C1, seqB; 9 PetscMPIInt size; 10 PetscErrorCode ierr; 11 PetscInt i,ia[2] = { 0, 2 }, ja[2] = { 0, 1 }, lda = 4; 12 PetscScalar a[2] = { 1.0, 1.0 }, *data; 13 PetscBool flg; 14 15 ierr = PetscInitialize(&argc, &argv, (char*)0,help);if (ierr) return ierr; 16 ierr = MPI_Comm_size(PETSC_COMM_WORLD, &size);CHKERRQ(ierr); 17 if (size != 2) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Must use 2 processors"); 18 19 ierr = MatCreate(PETSC_COMM_WORLD, &A);CHKERRQ(ierr); 20 ierr = MatSetType(A, MATMPIAIJ);CHKERRQ(ierr); 21 ierr = MatSetSizes(A, 1, 1, 2, 2);CHKERRQ(ierr); 22 ierr = MatMPIAIJSetPreallocationCSR(A, ia, ja, a);CHKERRQ(ierr); 23 24 ierr = PetscCalloc1(4 * lda,&data);CHKERRQ(ierr); 25 for (i = 0; i < 4; ++i) data[lda * i] = i * 1.0; 26 27 ierr = MatCreateDense(PETSC_COMM_WORLD, 1, PETSC_DECIDE, 2, 4, data, &B);CHKERRQ(ierr); 28 ierr = MatSetOptionsPrefix(B,"b_");CHKERRQ(ierr); 29 ierr = MatSetFromOptions(B);CHKERRQ(ierr); 30 ierr = MatDenseGetLocalMatrix(B, &seqB);CHKERRQ(ierr); 31 ierr = MatSeqDenseSetLDA(seqB, lda);CHKERRQ(ierr); 32 33 /* Test MatMatMult() */ 34 ierr = MatMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C);CHKERRQ(ierr); 35 ierr = MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C);CHKERRQ(ierr); 36 37 ierr = MatMatMultEqual(A,B,C,10,&flg);CHKERRQ(ierr); 38 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NOTSAMETYPE,"Error in MatMatMult() for C"); 39 40 /* Test user-provided mpidense matrix product */ 41 ierr = MatDuplicate(C,MAT_COPY_VALUES,&C1);CHKERRQ(ierr); 42 ierr = MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C1);CHKERRQ(ierr); 43 ierr = MatMatMultEqual(A,B,C1,10,&flg);CHKERRQ(ierr); 44 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NOTSAMETYPE,"Error in MatMatMult() for C1"); 45 46 ierr = MatDestroy(&C1);CHKERRQ(ierr); 47 ierr = MatDestroy(&C);CHKERRQ(ierr); 48 49 /* Test MatTransposeMatMult() */ 50 ierr = MatTransposeMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C);CHKERRQ(ierr); 51 ierr = MatTransposeMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C);CHKERRQ(ierr); 52 53 ierr = MatTransposeMatMultEqual(A,B,C,10,&flg);CHKERRQ(ierr); 54 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NOTSAMETYPE,"Error in MatTransposeMatMult()"); 55 ierr = MatDestroy(&C);CHKERRQ(ierr); 56 57 ierr = MatDestroy(&B);CHKERRQ(ierr); 58 ierr = MatDestroy(&A);CHKERRQ(ierr); 59 ierr = PetscFree(data);CHKERRQ(ierr); 60 61 ierr = PetscFinalize(); 62 return ierr; 63 } 64 65 /*TEST 66 67 test: 68 suffix: 1 69 nsize: 2 70 output_file: output/ex34.out 71 72 test: 73 suffix: 1_cuda 74 requires: cuda 75 nsize: 2 76 args: -b_mat_type mpidensecuda 77 output_file: output/ex34.out 78 TEST*/ 79