aij/seq/seqhashmat.h

26cec326SBarry Smith/*
26cec326SBarry Smith   used by SEQAIJ, BAIJ and SBAIJ to reduce code duplication
26cec326SBarry Smith
26cec326SBarry Smith     define TYPE to AIJ BAIJ or SBAIJ
26cec326SBarry Smith            TYPE_BS_ON for BAIJ and SBAIJ
26cec326SBarry Smith
26cec326SBarry Smith*/
26cec326SBarry Smithstatic PetscErrorCode MatAssemblyEnd_Seq_Hash(Mat A, MatAssemblyType type)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscConcat(Mat_Seq, TYPE) *a = (PetscConcat(Mat_Seq, TYPE) *)A->data;
26cec326SBarry Smith  PetscHashIter  hi;
26cec326SBarry Smith  PetscHashIJKey key;
26cec326SBarry Smith  PetscScalar    value, *values;
26cec326SBarry Smith  PetscInt       m, n, *cols, *rowstarts;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscInt bs;
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscCall(MatGetBlockSize(A, &bs));
26cec326SBarry Smith  if (bs > 1) PetscCall(PetscHSetIJDestroy(&a->bht));
26cec326SBarry Smith#endif
26cec326SBarry Smith  A->preallocated = PETSC_FALSE; /* this was set to true for the MatSetValues_Hash() to work */
26cec326SBarry Smith
aea10558SJacob Faibussowitsch  A->ops[0]      = a->cops;
ad79cf63SBarry Smith  A->hash_active = PETSC_FALSE;
26cec326SBarry Smith
26cec326SBarry Smith  /* move values from hash format to matrix type format */
26cec326SBarry Smith  PetscCall(MatGetSize(A, &m, NULL));
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  if (bs > 1) PetscCall(PetscConcat(PetscConcat(MatSeq, TYPE), SetPreallocation)(A, bs, PETSC_DETERMINE, a->bdnz));
26cec326SBarry Smith  else PetscCall(PetscConcat(PetscConcat(MatSeq, TYPE), SetPreallocation)(A, 1, PETSC_DETERMINE, a->dnz));
26cec326SBarry Smith#else
26cec326SBarry Smith  PetscCall(MatSeqAIJSetPreallocation(A, PETSC_DETERMINE, a->dnz));
26cec326SBarry Smith#endif
26cec326SBarry Smith  PetscCall(MatSetOption(A, MAT_NEW_NONZERO_ALLOCATION_ERR, PETSC_FALSE));
26cec326SBarry Smith  PetscCall(PetscHMapIJVGetSize(a->ht, &n));
*9f0612e4SBarry Smith  /* do not need PetscShmgetAllocateArray() since arrays are temporary */
26cec326SBarry Smith  PetscCall(PetscMalloc3(n, &cols, m + 1, &rowstarts, n, &values));
26cec326SBarry Smith  rowstarts[0] = 0;
26cec326SBarry Smith  for (PetscInt i = 0; i < m; i++) rowstarts[i + 1] = rowstarts[i] + a->dnz[i];
26cec326SBarry Smith
26cec326SBarry Smith  PetscHashIterBegin(a->ht, hi);
26cec326SBarry Smith  while (!PetscHashIterAtEnd(a->ht, hi)) {
26cec326SBarry Smith    PetscHashIterGetKey(a->ht, hi, key);
26cec326SBarry Smith    PetscHashIterGetVal(a->ht, hi, value);
26cec326SBarry Smith    cols[rowstarts[key.i]]     = key.j;
26cec326SBarry Smith    values[rowstarts[key.i]++] = value;
26cec326SBarry Smith    PetscHashIterNext(a->ht, hi);
26cec326SBarry Smith  }
26cec326SBarry Smith  PetscCall(PetscHMapIJVDestroy(&a->ht));
26cec326SBarry Smith
26cec326SBarry Smith  for (PetscInt i = 0, start = 0; i < m; i++) {
8e3a54c0SPierre Jolivet    PetscCall(MatSetValues(A, 1, &i, a->dnz[i], PetscSafePointerPlusOffset(cols, start), PetscSafePointerPlusOffset(values, start), A->insertmode));
26cec326SBarry Smith    start += a->dnz[i];
26cec326SBarry Smith  }
26cec326SBarry Smith  PetscCall(PetscFree3(cols, rowstarts, values));
26cec326SBarry Smith  PetscCall(PetscFree(a->dnz));
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  if (bs > 1) PetscCall(PetscFree(a->bdnz));
26cec326SBarry Smith#endif
26cec326SBarry Smith  PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith  PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smithstatic PetscErrorCode MatDestroy_Seq_Hash(Mat A)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscConcat(Mat_Seq, TYPE) *a = (PetscConcat(Mat_Seq, TYPE) *)A->data;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscInt bs;
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith  PetscCall(PetscHMapIJVDestroy(&a->ht));
26cec326SBarry Smith  PetscCall(PetscFree(a->dnz));
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscCall(MatGetBlockSize(A, &bs));
26cec326SBarry Smith  if (bs > 1) {
26cec326SBarry Smith    PetscCall(PetscFree(a->bdnz));
26cec326SBarry Smith    PetscCall(PetscHSetIJDestroy(&a->bht));
26cec326SBarry Smith  }
26cec326SBarry Smith#endif
26cec326SBarry Smith  PetscCall((*a->cops.destroy)(A));
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smithstatic PetscErrorCode MatZeroEntries_Seq_Hash(Mat A)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smithstatic PetscErrorCode MatSetRandom_Seq_Hash(Mat A, PetscRandom r)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith  SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONGSTATE, "Must set preallocation first");
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}
26cec326SBarry Smith
26cec326SBarry Smithstatic PetscErrorCode MatSetUp_Seq_Hash(Mat A)
26cec326SBarry Smith{
26cec326SBarry Smith  PetscConcat(Mat_Seq, TYPE) *a = (PetscConcat(Mat_Seq, TYPE) *)A->data;
26cec326SBarry Smith  PetscInt m;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscInt bs;
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith  PetscFunctionBegin;
26cec326SBarry Smith  PetscCall(PetscInfo(A, "Using hash-based MatSetValues() for MATSEQAIJ because no preallocation provided\n"));
26cec326SBarry Smith  PetscCall(PetscLayoutSetUp(A->rmap));
26cec326SBarry Smith  PetscCall(PetscLayoutSetUp(A->cmap));
26cec326SBarry Smith  if (A->rmap->bs < 1) A->rmap->bs = 1;
26cec326SBarry Smith  if (A->cmap->bs < 1) A->cmap->bs = 1;
26cec326SBarry Smith
26cec326SBarry Smith  PetscCall(MatGetLocalSize(A, &m, NULL));
26cec326SBarry Smith  PetscCall(PetscHMapIJVCreate(&a->ht));
26cec326SBarry Smith  PetscCall(PetscCalloc1(m, &a->dnz));
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  PetscCall(MatGetBlockSize(A, &bs));
26cec326SBarry Smith  if (bs > 1) {
26cec326SBarry Smith    PetscCall(PetscCalloc1(m / bs, &a->bdnz));
26cec326SBarry Smith    PetscCall(PetscHSetIJCreate(&a->bht));
26cec326SBarry Smith  }
26cec326SBarry Smith#endif
26cec326SBarry Smith
26cec326SBarry Smith  /* keep a record of the operations so they can be reset when the hash handling is complete */
aea10558SJacob Faibussowitsch  a->cops               = A->ops[0];
26cec326SBarry Smith  A->ops->assemblybegin = NULL;
26cec326SBarry Smith  A->ops->assemblyend   = MatAssemblyEnd_Seq_Hash;
26cec326SBarry Smith  A->ops->destroy       = MatDestroy_Seq_Hash;
26cec326SBarry Smith  A->ops->zeroentries   = MatZeroEntries_Seq_Hash;
26cec326SBarry Smith  A->ops->setrandom     = MatSetRandom_Seq_Hash;
26cec326SBarry Smith#if defined(TYPE_BS_ON)
26cec326SBarry Smith  if (bs > 1) A->ops->setvalues = MatSetValues_Seq_Hash_BS;
26cec326SBarry Smith  else
26cec326SBarry Smith#endif
26cec326SBarry Smith    A->ops->setvalues = MatSetValues_Seq_Hash;
26cec326SBarry Smith  A->ops->setvaluesblocked = NULL;
26cec326SBarry Smith
26cec326SBarry Smith  A->preallocated = PETSC_TRUE;
ad79cf63SBarry Smith  A->hash_active  = PETSC_TRUE;
26cec326SBarry Smith  PetscFunctionReturn(PETSC_SUCCESS);
26cec326SBarry Smith}