16679dcc1SBarry Smith 26679dcc1SBarry Smith static char help[] = "Creates MatSeqBAIJ matrix of given BS for timing tests of MatMult().\n"; 36679dcc1SBarry Smith 46679dcc1SBarry Smith #include <petscmat.h> 56679dcc1SBarry Smith 66679dcc1SBarry Smith int main(int argc,char **args) 76679dcc1SBarry Smith { 86679dcc1SBarry Smith Mat A; 96679dcc1SBarry Smith Vec x,y; 106679dcc1SBarry Smith PetscErrorCode ierr; 116679dcc1SBarry Smith PetscInt m=50000,bs=12,i,j,k,l,row,col,M; 126679dcc1SBarry Smith PetscScalar rval,*vals; 136679dcc1SBarry Smith PetscRandom rdm; 146679dcc1SBarry Smith 156679dcc1SBarry Smith ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 166679dcc1SBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-mat_block_size",&bs,NULL);CHKERRQ(ierr); 176679dcc1SBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-mat_size",&m,NULL);CHKERRQ(ierr); 186679dcc1SBarry Smith M = m*bs; 196679dcc1SBarry Smith ierr = MatCreateSeqBAIJ(PETSC_COMM_SELF,bs,M,M,27,NULL,&A);CHKERRQ(ierr); 20*e269983cSBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 216679dcc1SBarry Smith 226679dcc1SBarry Smith ierr = PetscRandomCreate(PETSC_COMM_SELF,&rdm);CHKERRQ(ierr); 236679dcc1SBarry Smith ierr = PetscRandomSetFromOptions(rdm);CHKERRQ(ierr); 246679dcc1SBarry Smith ierr = VecCreateSeq(PETSC_COMM_SELF,M,&x);CHKERRQ(ierr); 256679dcc1SBarry Smith ierr = VecDuplicate(x,&y);CHKERRQ(ierr); 266679dcc1SBarry Smith 27*e269983cSBarry Smith /* For each block row insert at most 27 blocks */ 286679dcc1SBarry Smith ierr = PetscMalloc1(bs*bs,&vals);CHKERRQ(ierr); 296679dcc1SBarry Smith for (i=0; i<m; i++) { 306679dcc1SBarry Smith row = i; 316679dcc1SBarry Smith for (j=0; j<27; j++) { 326679dcc1SBarry Smith ierr = PetscRandomGetValue(rdm,&rval);CHKERRQ(ierr); 336679dcc1SBarry Smith col = (PetscInt)(PetscRealPart(rval)*m); 346679dcc1SBarry Smith for (k=0; k<bs; k++) { 356679dcc1SBarry Smith for (l=0; l<bs; l++) { 366679dcc1SBarry Smith ierr = PetscRandomGetValue(rdm,&rval);CHKERRQ(ierr); 376679dcc1SBarry Smith vals[k*bs + l] = rval; 386679dcc1SBarry Smith } 396679dcc1SBarry Smith } 406679dcc1SBarry Smith ierr = MatSetValuesBlocked(A,1,&row,1,&col,vals,INSERT_VALUES);CHKERRQ(ierr); 416679dcc1SBarry Smith } 426679dcc1SBarry Smith } 436679dcc1SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 446679dcc1SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 456679dcc1SBarry Smith ierr = PetscFree(vals);CHKERRQ(ierr); 466679dcc1SBarry Smith 476679dcc1SBarry Smith /* Time MatMult(), MatMultAdd() */ 486679dcc1SBarry Smith for (i=0; i<25; i++) { 496679dcc1SBarry Smith ierr = VecSetRandom(x,rdm);CHKERRQ(ierr); 506679dcc1SBarry Smith ierr = MatMult(A,x,y);CHKERRQ(ierr); 516679dcc1SBarry Smith ierr = VecSetRandom(x,rdm);CHKERRQ(ierr); 526679dcc1SBarry Smith ierr = VecSetRandom(y,rdm);CHKERRQ(ierr); 536679dcc1SBarry Smith ierr = MatMultAdd(A,x,y,y);CHKERRQ(ierr); 546679dcc1SBarry Smith } 556679dcc1SBarry Smith 566679dcc1SBarry Smith ierr = MatDestroy(&A);CHKERRQ(ierr); 576679dcc1SBarry Smith ierr = VecDestroy(&x);CHKERRQ(ierr); 586679dcc1SBarry Smith ierr = VecDestroy(&y);CHKERRQ(ierr); 596679dcc1SBarry Smith ierr = PetscRandomDestroy(&rdm);CHKERRQ(ierr); 606679dcc1SBarry Smith ierr = PetscFinalize(); 616679dcc1SBarry Smith return ierr; 626679dcc1SBarry Smith } 636679dcc1SBarry Smith 646679dcc1SBarry Smith 656679dcc1SBarry Smith /*TEST 666679dcc1SBarry Smith 676679dcc1SBarry Smith test: 686679dcc1SBarry Smith args: -mat_block_size {{1 2 4 5 6 8 12 15}} 696679dcc1SBarry Smith 706679dcc1SBarry Smith TEST*/ 71