12eac72dbSBarry Smith /* 237f753daSBarry Smith Defines the vector component of PETSc. Vectors generally represent 337f753daSBarry Smith degrees of freedom for finite element/finite difference functions 484cb2905SBarry Smith on a grid. They have more mathematical structure then simple arrays. 52eac72dbSBarry Smith */ 62eac72dbSBarry Smith 797929ea7SJunchao Zhang #if !defined(PETSCVEC_H) 826bd1501SBarry Smith #define PETSCVEC_H 997929ea7SJunchao Zhang #include <petscsys.h> 1097929ea7SJunchao Zhang #include <petscsftypes.h> /* for VecScatter, VecScatterType */ 112c8e378dSBarry Smith #include <petscis.h> 12030f984aSJacob Faibussowitsch #include <petscdevicetypes.h> 13665c2dedSJed Brown #include <petscviewer.h> 142eac72dbSBarry Smith 1509321671SBarry Smith /*S 1609321671SBarry Smith Vec - Abstract PETSc vector object 1709321671SBarry Smith 1809321671SBarry Smith Level: beginner 1909321671SBarry Smith 2009321671SBarry Smith .seealso: VecCreate(), VecType, VecSetType() 2109321671SBarry Smith S*/ 22f09e8eb9SSatish Balay typedef struct _p_Vec* Vec; 2309321671SBarry Smith 2409321671SBarry Smith /*E 25398c84b2SBarry Smith ScatterMode - Determines the direction of a scatter 26398c84b2SBarry Smith 27398c84b2SBarry Smith Level: beginner 28398c84b2SBarry Smith 29398c84b2SBarry Smith .seealso: VecScatter, VecScatterBegin(), VecScatterEnd() 30398c84b2SBarry Smith E*/ 31398c84b2SBarry Smith typedef enum {SCATTER_FORWARD=0, SCATTER_REVERSE=1, SCATTER_FORWARD_LOCAL=2, SCATTER_REVERSE_LOCAL=3, SCATTER_LOCAL=2} ScatterMode; 32398c84b2SBarry Smith 33398c84b2SBarry Smith /*MC 34130e142eSJunchao Zhang SCATTER_FORWARD - Scatters the values as dictated by the VecScatterCreate() call 35398c84b2SBarry Smith 36398c84b2SBarry Smith Level: beginner 37398c84b2SBarry Smith 38130e142eSJunchao Zhang .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD_LOCAL, 39398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 40398c84b2SBarry Smith 41398c84b2SBarry Smith M*/ 42398c84b2SBarry Smith 43398c84b2SBarry Smith /*MC 44398c84b2SBarry Smith SCATTER_REVERSE - Moves the values in the opposite direction then the directions indicated in 45130e142eSJunchao Zhang in the VecScatterCreate() 46398c84b2SBarry Smith 47398c84b2SBarry Smith Level: beginner 48398c84b2SBarry Smith 49130e142eSJunchao Zhang .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 50398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 51398c84b2SBarry Smith 52398c84b2SBarry Smith M*/ 53398c84b2SBarry Smith 54398c84b2SBarry Smith /*MC 55130e142eSJunchao Zhang SCATTER_FORWARD_LOCAL - Scatters the values as dictated by the VecScatterCreate() call except NO parallel communication 56398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 57398c84b2SBarry Smith 58398c84b2SBarry Smith Level: developer 59398c84b2SBarry Smith 60130e142eSJunchao Zhang .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD, 61398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 62398c84b2SBarry Smith 63398c84b2SBarry Smith M*/ 64398c84b2SBarry Smith 65398c84b2SBarry Smith /*MC 66398c84b2SBarry Smith SCATTER_REVERSE_LOCAL - Moves the values in the opposite direction then the directions indicated in 67130e142eSJunchao Zhang in the VecScatterCreate() except NO parallel communication 68398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 69398c84b2SBarry Smith 70398c84b2SBarry Smith Level: developer 71398c84b2SBarry Smith 72130e142eSJunchao Zhang .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 73398c84b2SBarry Smith SCATTER_REVERSE 74398c84b2SBarry Smith 75398c84b2SBarry Smith M*/ 76398c84b2SBarry Smith 7776bdecfbSBarry Smith /*J 788f6c3df8SBarry Smith VecType - String with the name of a PETSc vector 7909321671SBarry Smith 8009321671SBarry Smith Level: beginner 8109321671SBarry Smith 828f6c3df8SBarry Smith .seealso: VecSetType(), Vec, VecCreate(), VecDestroy() 8376bdecfbSBarry Smith J*/ 8419fd82e9SBarry Smith typedef const char* VecType; 850676abe4SMatthew Knepley #define VECSEQ "seq" 860676abe4SMatthew Knepley #define VECMPI "mpi" 87f48c50deSBarry Smith #define VECSTANDARD "standard" /* seq on one process and mpi on several */ 880676abe4SMatthew Knepley #define VECSHARED "shared" 89b17c682bSKarl Rupp #define VECSEQVIENNACL "seqviennacl" 90b17c682bSKarl Rupp #define VECMPIVIENNACL "mpiviennacl" 91b17c682bSKarl Rupp #define VECVIENNACL "viennacl" /* seqviennacl on one process and mpiviennacl on several */ 9282f73ecaSAlejandro Lamas Daviña #define VECSEQCUDA "seqcuda" 9382f73ecaSAlejandro Lamas Daviña #define VECMPICUDA "mpicuda" 9482f73ecaSAlejandro Lamas Daviña #define VECCUDA "cuda" /* seqcuda on one process and mpicuda on several */ 9559af0bd3SScott Kruger #define VECSEQHIP "seqhip" 9659af0bd3SScott Kruger #define VECMPIHIP "mpihip" 9759af0bd3SScott Kruger #define VECHIP "hip" /* seqcuda on one process and mpicuda on several */ 981aae2881SJed Brown #define VECNEST "nest" 9910171e34SJunchao Zhang #define VECSEQKOKKOS "seqkokkos" 10010171e34SJunchao Zhang #define VECMPIKOKKOS "mpikokkos" 10110171e34SJunchao Zhang #define VECKOKKOS "kokkos" /* seqkokkos on one process and mpikokkos on several */ 1022eac72dbSBarry Smith 103803a1b88SHong Zhang /* Dynamic creation and loading functions */ 104803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetType(VecScatter, VecScatterType); 105803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterGetType(VecScatter, VecScatterType*); 106803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterSetFromOptions(VecScatter); 107803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecScatterRegister(const char[],PetscErrorCode (*)(VecScatter)); 108130e142eSJunchao Zhang PETSC_EXTERN PetscErrorCode VecScatterCreate(Vec,IS,Vec,IS,VecScatter*); 1094b71561bSShri Abhyankar 110fd487807SMatthew Knepley /* Logging support */ 111450a797fSBarry Smith #define REAL_FILE_CLASSID 1211213 1120700a824SBarry Smith #define VEC_FILE_CLASSID 1211214 113014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID; 11497929ea7SJunchao Zhang PETSC_EXTERN PetscClassId PETSCSF_CLASSID; 11545b63f25SDmitry Karpeev 116607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void); 117014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void); 118fd487807SMatthew Knepley 119014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*); 120014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*); 121014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*); 122014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*); 123014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*); 124014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*); 125803a1b88SHong Zhang PETSC_EXTERN PetscErrorCode VecCreateNode(MPI_Comm,PetscInt,PetscInt,Vec*); 126803a1b88SHong Zhang 127014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec); 128fe2efc57SMark PETSC_EXTERN PetscErrorCode VecViewFromOptions(Vec,PetscObject,const char[]); 1290d2bece7SBarry Smith 130014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec); 131014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*); 132014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec); 133014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]); 134014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]); 135014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]); 136f69a0ea3SMatthew Knepley 137014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt); 138fd487807SMatthew Knepley 13952b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*); 140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*); 14167392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*); 142014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*); 143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]); 144014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]); 145014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*); 146014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*); 147598f99faSAlp Dener PETSC_EXTERN PetscErrorCode VecConcatenate(PetscInt,const Vec[],Vec*,IS*[]); 148cddf8d76SBarry Smith 14909321671SBarry Smith /*E 15009321671SBarry Smith NormType - determines what type of norm to compute 15109321671SBarry Smith 15209321671SBarry Smith Level: beginner 15309321671SBarry Smith 15409321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm() 15509321671SBarry Smith E*/ 156*a873a8cdSSam Reynolds typedef enum {NORM_1=0, 157*a873a8cdSSam Reynolds NORM_2=1, 158*a873a8cdSSam Reynolds NORM_FROBENIUS=2, 159*a873a8cdSSam Reynolds NORM_INFINITY=3, 160*a873a8cdSSam Reynolds NORM_1_AND_2=4 161*a873a8cdSSam Reynolds } NormType; 1626a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[]; 163cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY 16409321671SBarry Smith 1659b250c83SBarry Smith /*MC 1669b250c83SBarry Smith NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum 1679b250c83SBarry Smith 1689b250c83SBarry Smith Level: beginner 1699b250c83SBarry Smith 1709b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS, 1719b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1729b250c83SBarry Smith 1739b250c83SBarry Smith M*/ 1749b250c83SBarry Smith 1759b250c83SBarry Smith /*MC 176df54d55cSMatthias Liertzer NORM_2 - the two norm, ||v|| = sqrt(sum_i |v_i|^2) (vectors only) 1779b250c83SBarry Smith 1789b250c83SBarry Smith Level: beginner 1799b250c83SBarry Smith 1809b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS, 1819b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1829b250c83SBarry Smith 1839b250c83SBarry Smith M*/ 1849b250c83SBarry Smith 1859b250c83SBarry Smith /*MC 186df54d55cSMatthias Liertzer NORM_FROBENIUS - ||A|| = sqrt(sum_ij |A_ij|^2), same as NORM_2 for vectors 1879b250c83SBarry Smith 1889b250c83SBarry Smith Level: beginner 1899b250c83SBarry Smith 1909b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1919b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1929b250c83SBarry Smith 1939b250c83SBarry Smith M*/ 1949b250c83SBarry Smith 1959b250c83SBarry Smith /*MC 1969b250c83SBarry Smith NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum 1979b250c83SBarry Smith 1989b250c83SBarry Smith Level: beginner 1999b250c83SBarry Smith 2009b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 20152e318dbSPierre Jolivet NORM_FROBENIUS, NORM_1_AND_2 2029b250c83SBarry Smith 2039b250c83SBarry Smith M*/ 2049b250c83SBarry Smith 2059b250c83SBarry Smith /*MC 2069b250c83SBarry Smith NORM_1_AND_2 - computes both the 1 and 2 norm of a vector 2079b250c83SBarry Smith 2089b250c83SBarry Smith Level: beginner 2099b250c83SBarry Smith 2109b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 21152e318dbSPierre Jolivet NORM_FROBENIUS, NORM_INFINITY 2129b250c83SBarry Smith 2139b250c83SBarry Smith M*/ 2149b250c83SBarry Smith 2159b250c83SBarry Smith /*MC 2169b250c83SBarry Smith NORM_MAX - see NORM_INFINITY 2179b250c83SBarry Smith 218d41222bbSBarry Smith Level: beginner 219d41222bbSBarry Smith 2209b250c83SBarry Smith M*/ 2219b250c83SBarry Smith 222*a873a8cdSSam Reynolds /*E 223*a873a8cdSSam Reynolds ReductionType - determines what type of column reduction to compute 224*a873a8cdSSam Reynolds 225*a873a8cdSSam Reynolds Level: beginner 226*a873a8cdSSam Reynolds 227*a873a8cdSSam Reynolds .seealso: MatGetColumnReductions(), MatGetColumnNorms(), NormType 228*a873a8cdSSam Reynolds E*/ 229*a873a8cdSSam Reynolds /* NOTE: If a new norm type is added, a corresponding ReductionType entry should be added. */ 230*a873a8cdSSam Reynolds typedef enum {REDUCTION_NORM_1=0, 231*a873a8cdSSam Reynolds REDUCTION_NORM_2=1, 232*a873a8cdSSam Reynolds REDUCTION_NORM_FROBENIUS=2, 233*a873a8cdSSam Reynolds REDUCTION_NORM_INFINITY=3, 234*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2=4, 235*a873a8cdSSam Reynolds REDUCTION_SUM=5, 236*a873a8cdSSam Reynolds REDUCTION_MEAN=6 237*a873a8cdSSam Reynolds } ReductionType; 238*a873a8cdSSam Reynolds 239*a873a8cdSSam Reynolds /*MC 240*a873a8cdSSam Reynolds REDUCTION_NORM_1 - corresponds to the NORM_1 NormType 241*a873a8cdSSam Reynolds 242*a873a8cdSSam Reynolds Level: beginner 243*a873a8cdSSam Reynolds 244*a873a8cdSSam Reynolds .seealso: ReductionType, NormType, MatGetColumnReductions(), MatGetColumnNorms(), 245*a873a8cdSSam Reynolds NORM_1, REDUCTION_NORM_2, REDUCTION_NORM_FROBENIUS, REDUCTION_NORM_INFINITY, 246*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2, REDUCTION_SUM, REDUCTION_MEAN 247*a873a8cdSSam Reynolds 248*a873a8cdSSam Reynolds M*/ 249*a873a8cdSSam Reynolds 250*a873a8cdSSam Reynolds /*MC 251*a873a8cdSSam Reynolds REDUCTION_NORM_2 - corresponds to the NORM_2 NormType 252*a873a8cdSSam Reynolds 253*a873a8cdSSam Reynolds Level: beginner 254*a873a8cdSSam Reynolds 255*a873a8cdSSam Reynolds .seealso: ReductionType, NormType, MatGetColumnReductions(), MatGetColumnNorms(), 256*a873a8cdSSam Reynolds NORM_2, REDUCTION_NORM_1, REDUCTION_NORM_FROBENIUS, REDUCTION_NORM_INFINITY, 257*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2, REDUCTION_SUM, REDUCTION_MEAN 258*a873a8cdSSam Reynolds 259*a873a8cdSSam Reynolds M*/ 260*a873a8cdSSam Reynolds 261*a873a8cdSSam Reynolds /*MC 262*a873a8cdSSam Reynolds REDUCTION_NORM_FROBENIUS - corresponds to the NORM_FROBENIUS NormType 263*a873a8cdSSam Reynolds 264*a873a8cdSSam Reynolds Level: beginner 265*a873a8cdSSam Reynolds 266*a873a8cdSSam Reynolds .seealso: ReductionType, NormType, MatGetColumnReductions(), MatGetColumnNorms(), 267*a873a8cdSSam Reynolds NORM_FROBENIUS, REDUCTION_NORM_1, REDUCTION_NORM_2, REDUCTION_NORM_INFINITY, 268*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2, REDUCTION_SUM, REDUCTION_MEAN 269*a873a8cdSSam Reynolds 270*a873a8cdSSam Reynolds M*/ 271*a873a8cdSSam Reynolds 272*a873a8cdSSam Reynolds /*MC 273*a873a8cdSSam Reynolds REDUCTION_NORM_INFINITY - corresponds to the NORM_INFINITY NormType 274*a873a8cdSSam Reynolds 275*a873a8cdSSam Reynolds Level: beginner 276*a873a8cdSSam Reynolds 277*a873a8cdSSam Reynolds .seealso: ReductionType, NormType, MatGetColumnReductions(), MatGetColumnNorms(), 278*a873a8cdSSam Reynolds NORM_INFINITY, REDUCTION_NORM_1, REDUCTION_NORM_2, REDUCTION_NORM_FROBENIUS, 279*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2, REDUCTION_SUM, REDUCTION_MEAN 280*a873a8cdSSam Reynolds 281*a873a8cdSSam Reynolds M*/ 282*a873a8cdSSam Reynolds 283*a873a8cdSSam Reynolds /*MC 284*a873a8cdSSam Reynolds REDUCTION_NORM_1_AND_2 - corresponds to the NORM_1_AND_2 NormType 285*a873a8cdSSam Reynolds 286*a873a8cdSSam Reynolds Level: beginner 287*a873a8cdSSam Reynolds 288*a873a8cdSSam Reynolds .seealso: ReductionType, NormType, MatGetColumnReductions(), MatGetColumnNorms(), 289*a873a8cdSSam Reynolds NORM_1_AND_2, REDUCTION_NORM_1, REDUCTION_NORM_2, REDUCTION_NORM_FROBENIUS, 290*a873a8cdSSam Reynolds REDUCTION_NORM_INFINITY, REDUCTION_SUM, REDUCTION_MEAN 291*a873a8cdSSam Reynolds 292*a873a8cdSSam Reynolds M*/ 293*a873a8cdSSam Reynolds 294*a873a8cdSSam Reynolds /*MC 295*a873a8cdSSam Reynolds REDUCTION_NORM_MAX - see REDUCTION_NORM_INFINITY 296*a873a8cdSSam Reynolds 297*a873a8cdSSam Reynolds Level: beginner 298*a873a8cdSSam Reynolds 299*a873a8cdSSam Reynolds .seealso: REDUCTION_NORM_INFINITY 300*a873a8cdSSam Reynolds 301*a873a8cdSSam Reynolds M*/ 302*a873a8cdSSam Reynolds 303*a873a8cdSSam Reynolds /*MC 304*a873a8cdSSam Reynolds REDUCTION_SUM - matrix column sum 305*a873a8cdSSam Reynolds 306*a873a8cdSSam Reynolds Level: beginner 307*a873a8cdSSam Reynolds 308*a873a8cdSSam Reynolds .seealso: ReductionType, MatGetColumnReductions(), REDUCTION_MEAN, REDUCTION_NORM_1, 309*a873a8cdSSam Reynolds REDUCTION_NORM_2, REDUCTION_NORM_FROBENIUS, REDUCTION_NORM_INFINITY 310*a873a8cdSSam Reynolds 311*a873a8cdSSam Reynolds M*/ 312*a873a8cdSSam Reynolds 313*a873a8cdSSam Reynolds /*MC 314*a873a8cdSSam Reynolds REDUCTION_MEAN - matrix column mean (arithmetic) 315*a873a8cdSSam Reynolds 316*a873a8cdSSam Reynolds Level: beginner 317*a873a8cdSSam Reynolds 318*a873a8cdSSam Reynolds .seealso: ReductionType, MatGetColumnReductions(), REDUCTION_SUM, REDUCTION_NORM_1, 319*a873a8cdSSam Reynolds REDUCTION_NORM_2, REDUCTION_NORM_FROBENIUS, REDUCTION_NORM_INFINITY 320*a873a8cdSSam Reynolds 321*a873a8cdSSam Reynolds M*/ 322*a873a8cdSSam Reynolds 323014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *); 324014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *); 325014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *); 326014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*); 327014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *); 328014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *); 329014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar); 330014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec); 331014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom); 332014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar); 333422a814eSBarry Smith PETSC_EXTERN PetscErrorCode VecSetInf(Vec); 334014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec); 335014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec); 336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec); 337014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]); 338014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec); 339014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec); 340014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec); 341014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec); 342014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec); 343014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec); 344014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec); 345014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec); 346014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*); 347014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar); 348014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec); 349014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool); 350014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec); 351014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec); 352014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec); 353014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec); 354014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*); 355014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]); 356014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]); 357014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]); 358014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []); 359014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []); 360014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]); 361849b11cfSMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecUniqueEntries(Vec,PetscInt*,PetscScalar**); 3624a560884SBarry Smith 363014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*); 364014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *); 365014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *); 366014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar); 367014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar); 368954b3ec6SBarry Smith 369014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode); 370014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode); 371014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode); 372014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode); 373d2655a18SBarry Smith 374bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetScatter(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 375bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetGather(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 376bdeb6c88SBarry Smith 377014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 378014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]); 379014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec); 380014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec); 381014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt); 382014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer); 383685405a1SBarry Smith PETSC_EXTERN PetscErrorCode VecStashViewFromOptions(Vec,PetscObject,const char[]); 384014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 38562dc5420SSatish Balay 38630de9b25SBarry Smith /*MC 38730de9b25SBarry Smith VecSetValue - Set a single entry into a vector. 38830de9b25SBarry Smith 38930de9b25SBarry Smith Synopsis: 390aaa7dc30SBarry Smith #include <petscvec.h> 3918122439eSJed Brown PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 39230de9b25SBarry Smith 39330de9b25SBarry Smith Not Collective 39430de9b25SBarry Smith 39530de9b25SBarry Smith Input Parameters: 39630de9b25SBarry Smith + v - the vector 39730de9b25SBarry Smith . row - the row location of the entry 39830de9b25SBarry Smith . value - the value to insert 39930de9b25SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 40030de9b25SBarry Smith 40130de9b25SBarry Smith Notes: 40230de9b25SBarry Smith For efficiency one should use VecSetValues() and set several or 40330de9b25SBarry Smith many values simultaneously if possible. 40430de9b25SBarry Smith 4051d73ed98SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 406879a43d7SBarry Smith MUST be called after all calls to VecSetValue() have been completed. 4071d73ed98SBarry Smith 408879a43d7SBarry Smith VecSetValue() uses 0-based indices in Fortran as well as in C. 4091d73ed98SBarry Smith 4101d73ed98SBarry Smith Level: beginner 4111d73ed98SBarry Smith 4121d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal() 4131d73ed98SBarry Smith M*/ 41450755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);} 4151d73ed98SBarry Smith 416014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt); 417014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*); 418014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 4198ed539a5SBarry Smith 420fd487807SMatthew Knepley /* Dynamic creation and loading functions */ 421140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList; 42219fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec,VecType); 42319fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec,VecType*); 424bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec)); 42530de9b25SBarry Smith 426014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode); 427014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode); 428014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*); 429246626dcSBarry Smith PETSC_EXTERN PetscErrorCode VecScatterSetUp(VecScatter); 430014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *); 431014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer); 432fe2efc57SMark PETSC_EXTERN PetscErrorCode VecScatterViewFromOptions(VecScatter,PetscObject,const char[]); 433d27e6124SSatish Balay PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt[],PetscInt[]); 434014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool*); 4352195c698SBarry Smith 436014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 437014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 438014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 439014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 440014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 441014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 442014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar*[]); 443014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar*[]); 444ab360428SBarry Smith 445fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 446fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 447fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray4dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 448fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray3dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 449fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray3dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 450fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray2dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 451fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray2dWrite(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 452fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray1dWrite(Vec,PetscInt,PetscInt,PetscScalar*[]); 453fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray1dWrite(Vec,PetscInt,PetscInt,PetscScalar*[]); 454fdc842d1SBarry Smith 4555edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 4565edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 4575edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 4585edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 4595edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 4605edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 4615edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray1dRead(Vec,PetscInt,PetscInt,PetscScalar*[]); 4625edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray1dRead(Vec,PetscInt,PetscInt,PetscScalar*[]); 4635edff71fSBarry Smith 464014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]); 465014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec); 466014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]); 467de0d48c1SKarl Rupp 468014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]); 469014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]); 47084cb2905SBarry Smith 471014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer); 472014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool*); 473014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec,PetscViewer); 4748ed539a5SBarry Smith 475014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*); 476014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*); 477014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*); 478014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt*[]); 4798ed539a5SBarry Smith 480014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 481014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 48288b03592SBarry Smith 4839a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDAGetArray(Vec,PetscScalar**); 4849a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDARestoreArray(Vec,PetscScalar**); 485de0d48c1SKarl Rupp 4869a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDAGetArrayRead(Vec,const PetscScalar**); 4879a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDARestoreArrayRead(Vec,const PetscScalar**); 488de0d48c1SKarl Rupp 4899a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDAGetArrayWrite(Vec,PetscScalar**); 4909a971ab9SStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDARestoreArrayWrite(Vec,PetscScalar**); 491de0d48c1SKarl Rupp 492cbf3516aSStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDAPlaceArray(Vec,const PetscScalar[]); 493cbf3516aSStefano Zampini PETSC_EXTERN PetscErrorCode VecCUDAReplaceArray(Vec,const PetscScalar[]); 494de0d48c1SKarl Rupp PETSC_EXTERN PetscErrorCode VecCUDAResetArray(Vec); 495de0d48c1SKarl Rupp 49659af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPGetArray(Vec,PetscScalar**); 49759af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPRestoreArray(Vec,PetscScalar**); 49859af0bd3SScott Kruger 49959af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPGetArrayRead(Vec,const PetscScalar**); 50059af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPRestoreArrayRead(Vec,const PetscScalar**); 50159af0bd3SScott Kruger 50259af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPGetArrayWrite(Vec,PetscScalar**); 50359af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPRestoreArrayWrite(Vec,PetscScalar**); 50459af0bd3SScott Kruger 50559af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPPlaceArray(Vec,const PetscScalar[]); 50659af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPReplaceArray(Vec,const PetscScalar[]); 50759af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecHIPResetArray(Vec); 50859af0bd3SScott Kruger 50976ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLGetCLContext(Vec, PETSC_UINTPTR_T*); 51076ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLGetCLQueue(Vec, PETSC_UINTPTR_T*); 51176ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLGetCLMemRead(Vec, PETSC_UINTPTR_T*); 51276ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLGetCLMemWrite(Vec, PETSC_UINTPTR_T*); 51376ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLRestoreCLMemWrite(Vec); 51476ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLGetCLMem(Vec, PETSC_UINTPTR_T*); 51576ec7f3eSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLRestoreCLMem(Vec); 51676ec7f3eSKaushik Kulkarni 51788b03592SBarry Smith /*MC 51888b03592SBarry Smith VecSetValueLocal - Set a single entry into a vector using the local numbering 51988b03592SBarry Smith 52088b03592SBarry Smith Synopsis: 521aaa7dc30SBarry Smith #include <petscvec.h> 5228122439eSJed Brown PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 52388b03592SBarry Smith 52488b03592SBarry Smith Not Collective 52588b03592SBarry Smith 52688b03592SBarry Smith Input Parameters: 52788b03592SBarry Smith + v - the vector 52888b03592SBarry Smith . row - the row location of the entry 52988b03592SBarry Smith . value - the value to insert 53088b03592SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 53188b03592SBarry Smith 53288b03592SBarry Smith Notes: 53388b03592SBarry Smith For efficiency one should use VecSetValues() and set several or 53488b03592SBarry Smith many values simultaneously if possible. 53588b03592SBarry Smith 53688b03592SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 53788b03592SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 53888b03592SBarry Smith 53988b03592SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 54088b03592SBarry Smith 54188b03592SBarry Smith Level: beginner 54288b03592SBarry Smith 54388b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue() 54488b03592SBarry Smith M*/ 54588b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);} 54688b03592SBarry Smith 547014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 548014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*); 54990f02eecSBarry Smith 550014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *); 551014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *); 552014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *); 553014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *); 554014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *); 555014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *); 556d3c178dbSBarry Smith 557014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 558014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 559014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 560014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 561014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm); 562a751f32aSSatish Balay 563b470e4b4SRichard Tran Mills PETSC_EXTERN PetscErrorCode VecBindToCPU(Vec,PetscBool); 564b470e4b4SRichard Tran Mills PETSC_DEPRECATED_FUNCTION("Use VecBindToCPU (since v3.13)") PETSC_STATIC_INLINE PetscErrorCode VecPinToCPU(Vec v,PetscBool flg) {return VecBindToCPU(v,flg);} 565387df023SRichard Tran Mills PETSC_EXTERN PetscErrorCode VecSetPinnedMemoryMin(Vec,size_t); 566387df023SRichard Tran Mills PETSC_EXTERN PetscErrorCode VecGetPinnedMemoryMin(Vec,size_t *); 567a751f32aSSatish Balay 5686fe358cdSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecGetOffloadMask(Vec,PetscOffloadMask *); 56948d97fd2SKaushik Kulkarni 570bda0d2a0SJed Brown typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES,VEC_SUBSET_OFF_PROC_ENTRIES} VecOption; 571014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool); 57290f02eecSBarry Smith 5735c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**); 574fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecGetArrayWrite(Vec,PetscScalar**); 5755c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**); 5765c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**); 577fdc842d1SBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArrayWrite(Vec,PetscScalar**); 5785c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**); 579046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVector(Vec,Vec); 580046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVector(Vec,Vec); 581046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVectorRead(Vec,Vec); 582046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVectorRead(Vec,Vec); 583a256111fSJunchao Zhang PETSC_EXTERN PetscErrorCode VecGetArrayAndMemType(Vec,PetscScalar**,PetscMemType*); 584a256111fSJunchao Zhang PETSC_EXTERN PetscErrorCode VecRestoreArrayAndMemType(Vec,PetscScalar**); 585a256111fSJunchao Zhang PETSC_EXTERN PetscErrorCode VecGetArrayReadAndMemType(Vec,const PetscScalar**,PetscMemType*); 586a256111fSJunchao Zhang PETSC_EXTERN PetscErrorCode VecRestoreArrayReadAndMemType(Vec,const PetscScalar**); 58710171e34SJunchao Zhang 5889521ec69SBarry Smith /*@C 5899521ec69SBarry Smith VecGetArrayPair - Accesses a pair of pointers for two vectors that may be common. When not common the first is read only 5909521ec69SBarry Smith 591d083f849SBarry Smith Logically Collective on x 5929521ec69SBarry Smith 593f5f57ec0SBarry Smith Input Parameters: 5949521ec69SBarry Smith + x - the vector 5959521ec69SBarry Smith - y - the second vector 5969521ec69SBarry Smith 597f5f57ec0SBarry Smith Output Parameters: 5989521ec69SBarry Smith + xv - location to put pointer to the first array 5999521ec69SBarry Smith - yv - location to put pointer to the second array 6009521ec69SBarry Smith 6019521ec69SBarry Smith Level: developer 6029521ec69SBarry Smith 603f5f57ec0SBarry Smith Not available from Fortran 604f5f57ec0SBarry Smith 6059521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecRestoreArrayPair() 6069521ec69SBarry Smith 6079521ec69SBarry Smith @*/ 608d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecGetArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 609d9ca1df4SBarry Smith { 610d9ca1df4SBarry Smith PetscErrorCode ierr; 611d9ca1df4SBarry Smith 612d9ca1df4SBarry Smith PetscFunctionBegin; 613d9ca1df4SBarry Smith ierr = VecGetArray(y,yv);CHKERRQ(ierr); 614d9ca1df4SBarry Smith if (x != y) { 615d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 616d9ca1df4SBarry Smith } else { 617d9ca1df4SBarry Smith *xv = *yv; 618d9ca1df4SBarry Smith } 619d9ca1df4SBarry Smith PetscFunctionReturn(0); 620d9ca1df4SBarry Smith } 6219521ec69SBarry Smith 6229521ec69SBarry Smith /*@C 6239521ec69SBarry Smith VecRestoreArrayPair - Returns a pair of pointers for two vectors that may be common. When not common the first is read only 6249521ec69SBarry Smith 625d083f849SBarry Smith Logically Collective on x 6269521ec69SBarry Smith 627f5f57ec0SBarry Smith Input Parameters: 6289521ec69SBarry Smith + x - the vector 6299521ec69SBarry Smith - y - the second vector 6309521ec69SBarry Smith 631f5f57ec0SBarry Smith Output Parameters: 6329521ec69SBarry Smith + xv - location to put pointer to the first array 6339521ec69SBarry Smith - yv - location to put pointer to the second array 6349521ec69SBarry Smith 6359521ec69SBarry Smith Level: developer 6369521ec69SBarry Smith 637f5f57ec0SBarry Smith Not available from Fortran 638f5f57ec0SBarry Smith 6399521ec69SBarry Smith .seealso: VecGetArray(), VecGetArrayRead(), VecGetArrayPair() 6409521ec69SBarry Smith 6419521ec69SBarry Smith @*/ 642d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecRestoreArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 643d9ca1df4SBarry Smith { 644d9ca1df4SBarry Smith PetscErrorCode ierr; 645d9ca1df4SBarry Smith 646d9ca1df4SBarry Smith PetscFunctionBegin; 647d9ca1df4SBarry Smith ierr = VecRestoreArray(y,yv);CHKERRQ(ierr); 648d9ca1df4SBarry Smith if (x != y) { 649d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 650d9ca1df4SBarry Smith } 651d9ca1df4SBarry Smith PetscFunctionReturn(0); 652d9ca1df4SBarry Smith } 653d9ca1df4SBarry Smith 654d9ca1df4SBarry Smith #if defined(PETSC_USE_DEBUG) 65538fecf7cSJunchao Zhang PETSC_EXTERN PetscErrorCode VecLockReadPush(Vec); 65638fecf7cSJunchao Zhang PETSC_EXTERN PetscErrorCode VecLockReadPop(Vec); 65738fecf7cSJunchao Zhang /* We also have a non-public VecLockWriteSet_Private() in vecimpl.h */ 658d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockGet(Vec,PetscInt*); 659da1c2f70SJunchao Zhang PETSC_STATIC_INLINE PetscErrorCode VecSetErrorIfLocked(Vec x,PetscInt arg) 660da1c2f70SJunchao Zhang { 661da1c2f70SJunchao Zhang PetscInt state; 662da1c2f70SJunchao Zhang PetscErrorCode ierr; 66338fecf7cSJunchao Zhang 664da1c2f70SJunchao Zhang PetscFunctionBegin; 665da1c2f70SJunchao Zhang ierr = VecLockGet(x,&state);CHKERRQ(ierr); 666da1c2f70SJunchao Zhang if (state != 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE," Vec is already locked for read-only or read/write access, argument # %d",arg); 667da1c2f70SJunchao Zhang PetscFunctionReturn(0); 668da1c2f70SJunchao Zhang } 66938fecf7cSJunchao Zhang /* The three are deprecated */ 6705f1efc1aSLisandro Dalcin PETSC_EXTERN PETSC_DEPRECATED_FUNCTION("Use VecLockReadPush() (since version 3.11)") PetscErrorCode VecLockPush(Vec); 6715f1efc1aSLisandro Dalcin PETSC_EXTERN PETSC_DEPRECATED_FUNCTION("Use VecLockReadPop() (since version 3.11)") PetscErrorCode VecLockPop(Vec); 672ed274036SLisandro Dalcin #define VecLocked(x,arg) VecSetErrorIfLocked(x,arg) PETSC_DEPRECATED_MACRO("GCC warning \"Use VecSetErrorIfLocked() (since version 3.11)\"") 673d9ca1df4SBarry Smith #else 67445fccdb2SJunchao Zhang #define VecLockReadPush(x) 0 67545fccdb2SJunchao Zhang #define VecLockReadPop(x) 0 67638fecf7cSJunchao Zhang #define VecLockGet(x,s) *(s) = 0 67745fccdb2SJunchao Zhang #define VecSetErrorIfLocked(x,arg) 0 67838fecf7cSJunchao Zhang /* The three are deprecated */ 67945fccdb2SJunchao Zhang #define VecLockPush(x) 0 68045fccdb2SJunchao Zhang #define VecLockPop(x) 0 68145fccdb2SJunchao Zhang #define VecLocked(x,arg) 0 682d9ca1df4SBarry Smith #endif 683e1fa1e0fSSatish Balay 68462796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool); 685f7ecc322SBarry Smith 68615091d37SBarry Smith /* 68715091d37SBarry Smith These numbers need to match the entries in 6883c94ec11SBarry Smith the function table in vecimpl.h 68915091d37SBarry Smith */ 69092d03f5aSMatthew G. Knepley typedef enum { VECOP_DUPLICATE = 0, VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_VIEWNATIVE = 68, VECOP_LOADNATIVE = 69 } VecOperation; 691014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void)); 692b19c1e4cSBarry Smith 693e182c471SBarry Smith /* 694e182c471SBarry Smith Routines for dealing with ghosted vectors: 695e182c471SBarry Smith vectors with ghost elements at the end of the array. 696e182c471SBarry Smith */ 697014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]); 698014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 699014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 700014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 701014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 702014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*); 703014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*); 7043efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*); 705014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 706014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 707e182c471SBarry Smith 708014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec); 709c92e3469SBarry Smith PETSC_EXTERN PetscErrorCode VecImaginaryPart(Vec); 710c92e3469SBarry Smith PETSC_EXTERN PetscErrorCode VecRealPart(Vec); 71134233285SBarry Smith 712014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*); 713014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*); 714bba1ac68SSatish Balay 7154473680cSBarry Smith PETSC_EXTERN PetscErrorCode ISComplementVec(IS,Vec,IS*); 7164473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecPow(Vec, PetscScalar); 7174473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecMedian(Vec, Vec, Vec, Vec); 718ce902467SBarry Smith PETSC_EXTERN PetscErrorCode VecWhichInactive(Vec, Vec, Vec, Vec, PetscBool, IS *); 7194473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetween(Vec, Vec, Vec, IS *); 7204473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetweenOrEqual(Vec, Vec, Vec, IS *); 7214473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichGreaterThan(Vec, Vec, IS *); 7224473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichLessThan(Vec, Vec, IS *); 7234473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichEqual(Vec, Vec, IS *); 7244473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISAXPY(Vec, IS, PetscScalar,Vec); 7258883cd90SMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecISCopy(Vec, IS, ScatterMode, Vec); 7264473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISSet(Vec,IS, PetscScalar); 7274473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecBoundGradientProjection(Vec, Vec, Vec, Vec, Vec); 7284473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepBoundInfo(Vec,Vec,Vec,Vec,PetscReal*, PetscReal*,PetscReal*); 7294473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMax(Vec, Vec, PetscReal *); 7304473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMaxBounded(Vec,Vec,Vec,Vec,PetscReal*); 7314473680cSBarry Smith 732014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec); 733014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec); 7347dbadf16SMatthew Knepley 735d59c15a7SBarry Smith /*S 736d59c15a7SBarry Smith Vecs - Collection of vectors where the data for the vectors is stored in 737759e7b9cSHong Zhang one contiguous memory 738d59c15a7SBarry Smith 739d59c15a7SBarry Smith Level: advanced 740d59c15a7SBarry Smith 741d59c15a7SBarry Smith Notes: 742d59c15a7SBarry Smith Temporary construct for handling multiply right hand side solves 743d59c15a7SBarry Smith 744d59c15a7SBarry Smith This is faked by storing a single vector that has enough array space for 745d59c15a7SBarry Smith n vectors 746d59c15a7SBarry Smith 747d59c15a7SBarry Smith S*/ 74895fbd943SSatish Balay struct _n_Vecs {PetscInt n; Vec v;}; 74995fbd943SSatish Balay typedef struct _n_Vecs* Vecs; 750e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs); 751e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*); 752e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*); 753e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*); 754e9fa29b7SSatish Balay 755e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 756b17c682bSKarl Rupp typedef struct _p_PetscViennaCLIndices* PetscViennaCLIndices; 757b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscViennaCLIndices*); 758b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesDestroy(PetscViennaCLIndices*); 759b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyToGPUSome_Public(Vec,PetscViennaCLIndices); 760b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyFromGPUSome_Public(Vec,PetscViennaCLIndices); 761b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCL(MPI_Comm,PetscInt,Vec*); 762b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCL(MPI_Comm,PetscInt,PetscInt,Vec*); 763b8ced49eSKarl Rupp #endif 76459af0bd3SScott Kruger #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 76559af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec); 76659af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 76759af0bd3SScott Kruger #endif 768fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA) 76982f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDA(MPI_Comm,PetscInt,Vec*); 77076d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDAWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,Vec*); 771f94e795aSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateSeqCUDAWithArrays(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,const PetscScalar*,Vec*); 77282f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDA(MPI_Comm,PetscInt,PetscInt,Vec*); 77376d6722aSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDAWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,Vec*); 774f94e795aSKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateMPICUDAWithArrays(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,const PetscScalar*,Vec*); 775b17c682bSKarl Rupp #endif 77659af0bd3SScott Kruger #if defined(PETSC_HAVE_HIP) 77759af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateSeqHIP(MPI_Comm,PetscInt,Vec*); 77859af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateSeqHIPWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,Vec*); 77959af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateSeqHIPWithArrays(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,const PetscScalar*,Vec*); 78059af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateMPIHIP(MPI_Comm,PetscInt,PetscInt,Vec*); 78159af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateMPIHIPWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,Vec*); 78259af0bd3SScott Kruger PETSC_EXTERN PetscErrorCode VecCreateMPIHIPWithArrays(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,const PetscScalar*,Vec*); 78359af0bd3SScott Kruger #endif 7841e8b6502SStefano Zampini #if defined(PETSC_HAVE_KOKKOS_KERNELS) 7851e8b6502SStefano Zampini PETSC_EXTERN PetscErrorCode VecCreateSeqKokkos(MPI_Comm,PetscInt,Vec*); 7861e8b6502SStefano Zampini PETSC_EXTERN PetscErrorCode VecCreateSeqKokkosWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar*,Vec*); 7871e8b6502SStefano Zampini PETSC_EXTERN PetscErrorCode VecCreateMPIKokkos(MPI_Comm,PetscInt,PetscInt,Vec*); 7881e8b6502SStefano Zampini PETSC_EXTERN PetscErrorCode VecCreateMPIKokkosWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar*,Vec*); 7891e8b6502SStefano Zampini #endif 790b17c682bSKarl Rupp 791014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**); 792014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*); 793014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*); 794014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec); 795014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*); 796014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*); 79745b63f25SDmitry Karpeev 798c5929fdfSBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsGetVec(PetscOptions,const char[],const char[],Vec,PetscBool*); 7994325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal); 800ab352700SBarry Smith 801077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*); 802077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout); 8039a6d0b0bSJed Brown 8049a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer); 8059a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **); 8069a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode); 80722d96a0cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscSectionVecNorm(PetscSection, PetscSection, Vec, NormType, PetscReal []); 8089a6d0b0bSJed Brown 8095543aaa0SToby Isaac /*S 8105543aaa0SToby Isaac VecTagger - Object used to manage the tagging of a subset of indices based on the values of a vector. The 8115543aaa0SToby Isaac motivating application is the selection of cells for refinement or coarsening based on vector containing 8125543aaa0SToby Isaac the values in an error indicator metric. 8135543aaa0SToby Isaac 8145543aaa0SToby Isaac Level: advanced 8155543aaa0SToby Isaac S*/ 81661bbf837SToby Isaac typedef struct _p_VecTagger *VecTagger; 8175543aaa0SToby Isaac 8185543aaa0SToby Isaac /*J 8195543aaa0SToby Isaac VecTaggerType - String with the name of a VecTagger type 8205543aaa0SToby Isaac 8215543aaa0SToby Isaac Level: advanced 8225543aaa0SToby Isaac J*/ 8235543aaa0SToby Isaac typedef const char* VecTaggerType; 8244178c1b7SToby Isaac /* tag where the vector values are in a box of explicitly defined values */ 825cab77d58SToby Isaac #define VECTAGGERABSOLUTE "absolute" 8264178c1b7SToby Isaac /* tag where the vector values are in a box of values relative to the set of all values in the vector */ 8275543aaa0SToby Isaac #define VECTAGGERRELATIVE "relative" 8285543aaa0SToby Isaac /* tag where the vector values are in a relative range of the *cumulative distribution* of values in the vector */ 8291d1d0dd8SToby Isaac #define VECTAGGERCDF "cdf" 8305543aaa0SToby Isaac /* tag a vector as the union of other tags */ 8315543aaa0SToby Isaac #define VECTAGGEROR "or" 8325543aaa0SToby Isaac /* tag a vector as the intersection of other tags */ 8335543aaa0SToby Isaac #define VECTAGGERAND "and" 8345543aaa0SToby Isaac 8355543aaa0SToby Isaac PETSC_EXTERN PetscClassId VEC_TAGGER_CLASSID; 83661bbf837SToby Isaac PETSC_EXTERN PetscFunctionList VecTaggerList; 83761bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRegister(const char[],PetscErrorCode (*) (VecTagger)); 8385543aaa0SToby Isaac 8395543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCreate(MPI_Comm,VecTagger *); 840cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetBlockSize(VecTagger,PetscInt); 841cab77d58SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetBlockSize(VecTagger,PetscInt*); 8425543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetType(VecTagger,VecTaggerType); 8435543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetType(VecTagger,VecTaggerType *); 8445543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetInvert(VecTagger,PetscBool); 8455543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerGetInvert(VecTagger,PetscBool*); 8465543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetFromOptions(VecTagger); 8475543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerSetUp(VecTagger); 8485543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerView(VecTagger,PetscViewer); 8495543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeIS(VecTagger,Vec,IS *); 8505543aaa0SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerDestroy(VecTagger *); 8515543aaa0SToby Isaac 8524178c1b7SToby Isaac /*S 8534178c1b7SToby Isaac VecTaggerBox - A box range used to tag values. For real scalars, this is just a closed interval; for complex scalars, the box is the closed region in the complex plane 8544178c1b7SToby Isaac such that real(min) <= real(z) <= real(max) and imag(min) <= imag(z) <= imag(max). INF is an acceptable endpoint. 85561bbf837SToby Isaac 8564178c1b7SToby Isaac Level: beginner 85761bbf837SToby Isaac 8584178c1b7SToby Isaac .seealso: VecTaggerComputeIntervals() 8594178c1b7SToby Isaac S*/ 8604178c1b7SToby Isaac typedef struct { 8614178c1b7SToby Isaac PetscScalar min; 8624178c1b7SToby Isaac PetscScalar max; 8634178c1b7SToby Isaac } VecTaggerBox; 8644178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerComputeBoxes(VecTagger,Vec,PetscInt *,VecTaggerBox **); 8654178c1b7SToby Isaac 8664178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteSetBox(VecTagger,VecTaggerBox *); 8674178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAbsoluteGetBox(VecTagger,const VecTaggerBox **); 8684178c1b7SToby Isaac 8694178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeSetBox(VecTagger,VecTaggerBox *); 8704178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerRelativeGetBox(VecTagger,const VecTaggerBox **); 8714178c1b7SToby Isaac 8724178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetBox(VecTagger,VecTaggerBox *); 8734178c1b7SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetBox(VecTagger,const VecTaggerBox **); 8741d1d0dd8SToby Isaac 8751d1d0dd8SToby Isaac /*E 8761d1d0dd8SToby Isaac VecTaggerCDFMethod - Determines what method is used to compute absolute values from cumulative distribution values (e.g., what value is the preimage of .95 in the cdf). Relevant only in parallel: in serial it is directly computed. 8771d1d0dd8SToby Isaac 8781d1d0dd8SToby Isaac Level: advanced 8791d1d0dd8SToby Isaac .seealso: VecTaggerCDFSetMethod(), VecTaggerCDFMethods 8801d1d0dd8SToby Isaac E*/ 8811d1d0dd8SToby Isaac typedef enum {VECTAGGER_CDF_GATHER,VECTAGGER_CDF_ITERATIVE,VECTAGGER_CDF_NUM_METHODS} VecTaggerCDFMethod; 8821d1d0dd8SToby Isaac PETSC_EXTERN const char *const VecTaggerCDFMethods[]; 8831d1d0dd8SToby Isaac 8841d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFSetMethod(VecTagger,VecTaggerCDFMethod); 8851d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFGetMethod(VecTagger,VecTaggerCDFMethod*); 8861d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeSetTolerances(VecTagger,PetscInt,PetscReal,PetscReal); 8871d1d0dd8SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerCDFIterativeGetTolerances(VecTagger,PetscInt*,PetscReal*,PetscReal*); 88861bbf837SToby Isaac 889497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode); 890497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerOrGetSubs(VecTagger,PetscInt*,VecTagger**); 89161bbf837SToby Isaac 892497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndSetSubs(VecTagger,PetscInt,VecTagger*,PetscCopyMode); 893497ee004SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerAndGetSubs(VecTagger,PetscInt*,VecTagger**); 89461bbf837SToby Isaac 89561bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerInitializePackage(void); 89661bbf837SToby Isaac PETSC_EXTERN PetscErrorCode VecTaggerFinalizePackage(void); 89761bbf837SToby Isaac 8982eac72dbSBarry Smith #endif 899