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 70a835dfdSSatish Balay #ifndef __PETSCVEC_H 80a835dfdSSatish Balay #define __PETSCVEC_H 92c8e378dSBarry Smith #include <petscis.h> 10665c2dedSJed Brown #include <petscviewer.h> 112eac72dbSBarry Smith 1209321671SBarry Smith /*S 1309321671SBarry Smith Vec - Abstract PETSc vector object 1409321671SBarry Smith 1509321671SBarry Smith Level: beginner 1609321671SBarry Smith 1709321671SBarry Smith Concepts: field variables, unknowns, arrays 1809321671SBarry Smith 1909321671SBarry Smith .seealso: VecCreate(), VecType, VecSetType() 2009321671SBarry Smith S*/ 21f09e8eb9SSatish Balay typedef struct _p_Vec* Vec; 2209321671SBarry Smith 2309321671SBarry Smith /*S 2409321671SBarry Smith VecScatter - Object used to manage communication of data 2509321671SBarry Smith between vectors in parallel. Manages both scatters and gathers 2609321671SBarry Smith 2709321671SBarry Smith Level: beginner 2809321671SBarry Smith 2909321671SBarry Smith Concepts: scatter 3009321671SBarry Smith 3109321671SBarry Smith .seealso: VecScatterCreate(), VecScatterBegin(), VecScatterEnd() 3209321671SBarry Smith S*/ 33f09e8eb9SSatish Balay typedef struct _p_VecScatter* VecScatter; 3409321671SBarry Smith 3509321671SBarry Smith /*E 36398c84b2SBarry Smith ScatterMode - Determines the direction of a scatter 37398c84b2SBarry Smith 38398c84b2SBarry Smith Level: beginner 39398c84b2SBarry Smith 40398c84b2SBarry Smith .seealso: VecScatter, VecScatterBegin(), VecScatterEnd() 41398c84b2SBarry Smith E*/ 42398c84b2SBarry Smith typedef enum {SCATTER_FORWARD=0, SCATTER_REVERSE=1, SCATTER_FORWARD_LOCAL=2, SCATTER_REVERSE_LOCAL=3, SCATTER_LOCAL=2} ScatterMode; 43398c84b2SBarry Smith 44398c84b2SBarry Smith /*MC 45398c84b2SBarry Smith SCATTER_FORWARD - Scatters the values as dictated by the VecScatterCreate() call 46398c84b2SBarry Smith 47398c84b2SBarry Smith Level: beginner 48398c84b2SBarry Smith 49398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD_LOCAL, 50398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 51398c84b2SBarry Smith 52398c84b2SBarry Smith M*/ 53398c84b2SBarry Smith 54398c84b2SBarry Smith /*MC 55398c84b2SBarry Smith SCATTER_REVERSE - Moves the values in the opposite direction then the directions indicated in 56398c84b2SBarry Smith in the VecScatterCreate() 57398c84b2SBarry Smith 58398c84b2SBarry Smith Level: beginner 59398c84b2SBarry Smith 60398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 61398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 62398c84b2SBarry Smith 63398c84b2SBarry Smith M*/ 64398c84b2SBarry Smith 65398c84b2SBarry Smith /*MC 66398c84b2SBarry Smith SCATTER_FORWARD_LOCAL - Scatters the values as dictated by the VecScatterCreate() call except NO parallel communication 67398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 68398c84b2SBarry Smith 69398c84b2SBarry Smith Level: developer 70398c84b2SBarry Smith 71398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_REVERSE, SCATTER_FORWARD, 72398c84b2SBarry Smith SCATTER_REVERSE_LOCAL 73398c84b2SBarry Smith 74398c84b2SBarry Smith M*/ 75398c84b2SBarry Smith 76398c84b2SBarry Smith /*MC 77398c84b2SBarry Smith SCATTER_REVERSE_LOCAL - Moves the values in the opposite direction then the directions indicated in 78398c84b2SBarry Smith in the VecScatterCreate() except NO parallel communication 79398c84b2SBarry Smith is done. Any variables that have be moved between processes are ignored 80398c84b2SBarry Smith 81398c84b2SBarry Smith Level: developer 82398c84b2SBarry Smith 83398c84b2SBarry Smith .seealso: VecScatter, ScatterMode, VecScatterCreate(), VecScatterBegin(), VecScatterEnd(), SCATTER_FORWARD, SCATTER_FORWARD_LOCAL, 84398c84b2SBarry Smith SCATTER_REVERSE 85398c84b2SBarry Smith 86398c84b2SBarry Smith M*/ 87398c84b2SBarry Smith 8876bdecfbSBarry Smith /*J 898f6c3df8SBarry Smith VecType - String with the name of a PETSc vector 9009321671SBarry Smith 9109321671SBarry Smith Level: beginner 9209321671SBarry Smith 938f6c3df8SBarry Smith .seealso: VecSetType(), Vec, VecCreate(), VecDestroy() 9476bdecfbSBarry Smith J*/ 9519fd82e9SBarry Smith typedef const char* VecType; 960676abe4SMatthew Knepley #define VECSEQ "seq" 970676abe4SMatthew Knepley #define VECMPI "mpi" 98f48c50deSBarry Smith #define VECSTANDARD "standard" /* seq on one process and mpi on several */ 990676abe4SMatthew Knepley #define VECSHARED "shared" 1008154be41SBarry Smith #define VECSEQCUSP "seqcusp" 1018154be41SBarry Smith #define VECMPICUSP "mpicusp" 1028154be41SBarry Smith #define VECCUSP "cusp" /* seqcusp on one process and mpicusp on several */ 103b17c682bSKarl Rupp #define VECSEQVIENNACL "seqviennacl" 104b17c682bSKarl Rupp #define VECMPIVIENNACL "mpiviennacl" 105b17c682bSKarl Rupp #define VECVIENNACL "viennacl" /* seqviennacl on one process and mpiviennacl on several */ 106*82f73ecaSAlejandro Lamas Daviña #define VECSEQCUDA "seqcuda" 107*82f73ecaSAlejandro Lamas Daviña #define VECMPICUDA "mpicuda" 108*82f73ecaSAlejandro Lamas Daviña #define VECCUDA "cuda" /* seqcuda on one process and mpicuda on several */ 1091aae2881SJed Brown #define VECNEST "nest" 1101bf2b38bSKerry Stevens #define VECSEQPTHREAD "seqpthread" 1114b71561bSShri Abhyankar #define VECMPIPTHREAD "mpipthread" 112bf2c1783SBarry Smith #define VECPTHREAD "pthread" /* seqpthread on one process and mpipthread on several */ 1132eac72dbSBarry Smith 1144b71561bSShri Abhyankar 115fd487807SMatthew Knepley /* Logging support */ 1160700a824SBarry Smith #define VEC_FILE_CLASSID 1211214 117014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_CLASSID; 118014dd563SJed Brown PETSC_EXTERN PetscClassId VEC_SCATTER_CLASSID; 1198ba1e511SMatthew Knepley 12045b63f25SDmitry Karpeev 121607a6623SBarry Smith PETSC_EXTERN PetscErrorCode VecInitializePackage(void); 122014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecFinalizePackage(void); 123fd487807SMatthew Knepley 124014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreate(MPI_Comm,Vec*); 125014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeq(MPI_Comm,PetscInt,Vec*); 126014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPI(MPI_Comm,PetscInt,PetscInt,Vec*); 127014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,const PetscScalar[],Vec*); 128014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPIWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscScalar[],Vec*); 129014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateShared(MPI_Comm,PetscInt,PetscInt,Vec*); 130014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetFromOptions(Vec); 131685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecViewFromOptions(Vec A,PetscObject B,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,B,name);} 1320d2bece7SBarry Smith 133014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetUp(Vec); 134014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroy(Vec*); 135014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecZeroEntries(Vec); 136014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOptionsPrefix(Vec,const char[]); 137014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAppendOptionsPrefix(Vec,const char[]); 138014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOptionsPrefix(Vec,const char*[]); 139f69a0ea3SMatthew Knepley 140014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetSizes(Vec,PetscInt,PetscInt); 141fd487807SMatthew Knepley 14252b6e47cSBarry Smith PETSC_EXTERN PetscErrorCode VecDotNorm2(Vec,Vec,PetscScalar*,PetscReal*); 143014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDot(Vec,Vec,PetscScalar*); 14467392de3SBarry Smith PETSC_EXTERN PetscErrorCode VecDotRealPart(Vec,Vec,PetscReal*); 145014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDot(Vec,Vec,PetscScalar*); 146014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDot(Vec,PetscInt,const Vec[],PetscScalar[]); 147014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDot(Vec,PetscInt,const Vec[],PetscScalar[]); 148014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSubVector(Vec,IS,Vec*); 149014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreSubVector(Vec,IS,Vec*); 150cddf8d76SBarry Smith 15109321671SBarry Smith /*E 15209321671SBarry Smith NormType - determines what type of norm to compute 15309321671SBarry Smith 15409321671SBarry Smith Level: beginner 15509321671SBarry Smith 15609321671SBarry Smith .seealso: VecNorm(), VecNormBegin(), VecNormEnd(), MatNorm() 15709321671SBarry Smith E*/ 1589dcbbd2bSBarry Smith typedef enum {NORM_1=0,NORM_2=1,NORM_FROBENIUS=2,NORM_INFINITY=3,NORM_1_AND_2=4} NormType; 1596a6fc655SJed Brown PETSC_EXTERN const char *const NormTypes[]; 160cddf8d76SBarry Smith #define NORM_MAX NORM_INFINITY 16109321671SBarry Smith 1629b250c83SBarry Smith /*MC 1639b250c83SBarry Smith NORM_1 - the one norm, ||v|| = sum_i | v_i |. ||A|| = max_j || v_*j ||, maximum column sum 1649b250c83SBarry Smith 1659b250c83SBarry Smith Level: beginner 1669b250c83SBarry Smith 1679b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_2, NORM_FROBENIUS, 1689b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1699b250c83SBarry Smith 1709b250c83SBarry Smith M*/ 1719b250c83SBarry Smith 1729b250c83SBarry Smith /*MC 1739b250c83SBarry Smith NORM_2 - the two norm, ||v|| = sqrt(sum_i (v_i)^2) (vectors only) 1749b250c83SBarry Smith 1759b250c83SBarry Smith Level: beginner 1769b250c83SBarry Smith 1779b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_FROBENIUS, 1789b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1799b250c83SBarry Smith 1809b250c83SBarry Smith M*/ 1819b250c83SBarry Smith 1829b250c83SBarry Smith /*MC 1839b250c83SBarry Smith NORM_FROBENIUS - ||A|| = sqrt(sum_ij (A_ij)^2), same as NORM_2 for vectors 1849b250c83SBarry Smith 1859b250c83SBarry Smith Level: beginner 1869b250c83SBarry Smith 1879b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1889b250c83SBarry Smith NORM_INFINITY, NORM_1_AND_2 1899b250c83SBarry Smith 1909b250c83SBarry Smith M*/ 1919b250c83SBarry Smith 1929b250c83SBarry Smith /*MC 1939b250c83SBarry Smith NORM_INFINITY - ||v|| = max_i |v_i|. ||A|| = max_i || v_i* ||, maximum row sum 1949b250c83SBarry Smith 1959b250c83SBarry Smith Level: beginner 1969b250c83SBarry Smith 1979b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 1989b250c83SBarry Smith NORM_FROBINIUS, NORM_1_AND_2 1999b250c83SBarry Smith 2009b250c83SBarry Smith M*/ 2019b250c83SBarry Smith 2029b250c83SBarry Smith /*MC 2039b250c83SBarry Smith NORM_1_AND_2 - computes both the 1 and 2 norm of a vector 2049b250c83SBarry Smith 2059b250c83SBarry Smith Level: beginner 2069b250c83SBarry Smith 2079b250c83SBarry Smith .seealso: NormType, MatNorm(), VecNorm(), VecNormBegin(), VecNormEnd(), NORM_1, NORM_2, 2089b250c83SBarry Smith NORM_FROBINIUS, NORM_INFINITY 2099b250c83SBarry Smith 2109b250c83SBarry Smith M*/ 2119b250c83SBarry Smith 2129b250c83SBarry Smith /*MC 2139b250c83SBarry Smith NORM_MAX - see NORM_INFINITY 2149b250c83SBarry Smith 215d41222bbSBarry Smith Level: beginner 216d41222bbSBarry Smith 2179b250c83SBarry Smith M*/ 2189b250c83SBarry Smith 219014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNorm(Vec,NormType,PetscReal *); 220014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormAvailable(Vec,NormType,PetscBool *,PetscReal *); 221014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormalize(Vec,PetscReal *); 222014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSum(Vec,PetscScalar*); 223014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMax(Vec,PetscInt*,PetscReal *); 224014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMin(Vec,PetscInt*,PetscReal *); 225014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScale(Vec,PetscScalar); 226014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCopy(Vec,Vec); 227014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetRandom(Vec,PetscRandom); 228014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSet(Vec,PetscScalar); 229422a814eSBarry Smith PETSC_EXTERN PetscErrorCode VecSetInf(Vec); 230014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSwap(Vec,Vec); 231014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPY(Vec,PetscScalar,Vec); 232014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBY(Vec,PetscScalar,PetscScalar,Vec); 233014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMAXPY(Vec,PetscInt,const PetscScalar[],Vec[]); 234014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAYPX(Vec,PetscScalar,Vec); 235014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecWAXPY(Vec,PetscScalar,Vec,Vec); 236014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAXPBYPCZ(Vec,PetscScalar,PetscScalar,PetscScalar,Vec,Vec); 237014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMax(Vec,Vec,Vec); 238014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMaxAbs(Vec,Vec,Vec); 239014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMin(Vec,Vec,Vec); 240014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseMult(Vec,Vec,Vec); 241014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPointwiseDivide(Vec,Vec,Vec); 242014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMaxPointwiseDivide(Vec,Vec,PetscReal*); 243014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecShift(Vec,PetscScalar); 244014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReciprocal(Vec); 245014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPermute(Vec, IS, PetscBool ); 246014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSqrtAbs(Vec); 247014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLog(Vec); 248014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecExp(Vec); 249014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAbs(Vec); 250014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicate(Vec,Vec*); 251014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDuplicateVecs(Vec,PetscInt,Vec*[]); 252014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDestroyVecs(PetscInt, Vec*[]); 253014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNormAll(Vec,NormType,PetscReal[]); 254014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMaxAll(Vec,PetscInt [],PetscReal []); 255014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMinAll(Vec,PetscInt [],PetscReal []); 256014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScaleAll(Vec,const PetscScalar[]); 257849b11cfSMatthew G. Knepley PETSC_EXTERN PetscErrorCode VecUniqueEntries(Vec,PetscInt*,PetscScalar**); 2584a560884SBarry Smith 259014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideNorm(Vec,PetscInt,NormType,PetscReal*); 260014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMax(Vec,PetscInt,PetscInt *,PetscReal *); 261014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideMin(Vec,PetscInt,PetscInt *,PetscReal *); 262014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScale(Vec,PetscInt,PetscScalar); 263014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideSet(Vec,PetscInt,PetscScalar); 264954b3ec6SBarry Smith 265954b3ec6SBarry Smith 266014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGather(Vec,PetscInt,Vec,InsertMode); 267014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatter(Vec,PetscInt,Vec,InsertMode); 268014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideGatherAll(Vec,Vec[],InsertMode); 269014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStrideScatterAll(Vec[],Vec,InsertMode); 270d2655a18SBarry Smith 271bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetScatter(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 272bdeb6c88SBarry Smith PETSC_EXTERN PetscErrorCode VecStrideSubSetGather(Vec,PetscInt,const PetscInt[],const PetscInt[],Vec,InsertMode); 273bdeb6c88SBarry Smith 274014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValues(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 275014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetValues(Vec,PetscInt,const PetscInt[],PetscScalar[]); 276014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyBegin(Vec); 277014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecAssemblyEnd(Vec); 278014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashSetInitialSize(Vec,PetscInt,PetscInt); 279014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashView(Vec,PetscViewer); 280685405a1SBarry Smith PETSC_EXTERN PetscErrorCode VecStashViewFromOptions(Vec,PetscObject,const char[]); 281014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecStashGetInfo(Vec,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 28262dc5420SSatish Balay 28330de9b25SBarry Smith /*MC 28430de9b25SBarry Smith VecSetValue - Set a single entry into a vector. 28530de9b25SBarry Smith 28630de9b25SBarry Smith Synopsis: 287aaa7dc30SBarry Smith #include <petscvec.h> 2888122439eSJed Brown PetscErrorCode VecSetValue(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 28930de9b25SBarry Smith 29030de9b25SBarry Smith Not Collective 29130de9b25SBarry Smith 29230de9b25SBarry Smith Input Parameters: 29330de9b25SBarry Smith + v - the vector 29430de9b25SBarry Smith . row - the row location of the entry 29530de9b25SBarry Smith . value - the value to insert 29630de9b25SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 29730de9b25SBarry Smith 29830de9b25SBarry Smith Notes: 29930de9b25SBarry Smith For efficiency one should use VecSetValues() and set several or 30030de9b25SBarry Smith many values simultaneously if possible. 30130de9b25SBarry Smith 3021d73ed98SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 3031d73ed98SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 3041d73ed98SBarry Smith 3051d73ed98SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 3061d73ed98SBarry Smith 3071d73ed98SBarry Smith Level: beginner 3081d73ed98SBarry Smith 3091d73ed98SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValueLocal() 3101d73ed98SBarry Smith M*/ 31150755921SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValue(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValues(v,1,&i,&va,mode);} 3121d73ed98SBarry Smith 31330de9b25SBarry Smith 314014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetBlockSize(Vec,PetscInt); 315014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetBlockSize(Vec,PetscInt*); 316014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlocked(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 3178ed539a5SBarry Smith 318fd487807SMatthew Knepley /* Dynamic creation and loading functions */ 319140e18c1SBarry Smith PETSC_EXTERN PetscFunctionList VecList; 32019fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecSetType(Vec, VecType); 32119fd82e9SBarry Smith PETSC_EXTERN PetscErrorCode VecGetType(Vec, VecType *); 322bdf89e91SBarry Smith PETSC_EXTERN PetscErrorCode VecRegister(const char[],PetscErrorCode (*)(Vec)); 32330de9b25SBarry Smith 324014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreate(Vec,IS,Vec,IS,VecScatter *); 325014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateEmpty(MPI_Comm,VecScatter *); 326014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateLocal(VecScatter,PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt,const PetscInt[],const PetscInt[],const PetscInt[],PetscInt); 327014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterBegin(VecScatter,Vec,Vec,InsertMode,ScatterMode); 328014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterEnd(VecScatter,Vec,Vec,InsertMode,ScatterMode); 329014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterDestroy(VecScatter*); 330014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCopy(VecScatter,VecScatter *); 331014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterView(VecScatter,PetscViewer); 332685405a1SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecScatterViewFromOptions(VecScatter A,PetscObject obj,const char name[]) {return PetscObjectViewFromOptions((PetscObject)A,obj,name);} 333014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterRemap(VecScatter,PetscInt *,PetscInt*); 334014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterGetMerged(VecScatter,PetscBool *); 3352195c698SBarry Smith 336014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 337014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray4d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 338014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 339014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray3d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 340014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 341014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray2d(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 342014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 343014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray1d(Vec,PetscInt,PetscInt,PetscScalar *[]); 344ab360428SBarry Smith 3455edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 3465edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray4dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar****[]); 3475edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 3485edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray3dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar***[]); 3495edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 3505edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray2dRead(Vec,PetscInt,PetscInt,PetscInt,PetscInt,PetscScalar**[]); 3515edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecGetArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]); 3525edff71fSBarry Smith PETSC_EXTERN PetscErrorCode VecRestoreArray1dRead(Vec,PetscInt,PetscInt,PetscScalar *[]); 3535edff71fSBarry Smith 354014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecPlaceArray(Vec,const PetscScalar[]); 355014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecResetArray(Vec); 356014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecReplaceArray(Vec,const PetscScalar[]); 357014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrays(const Vec[],PetscInt,PetscScalar**[]); 358014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrays(const Vec[],PetscInt,PetscScalar**[]); 35984cb2905SBarry Smith 360014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecView(Vec,PetscViewer); 361014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecEqual(Vec,Vec,PetscBool *); 362014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecLoad(Vec, PetscViewer); 3638ed539a5SBarry Smith 364014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetSize(Vec,PetscInt*); 365014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalSize(Vec,PetscInt*); 366014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRange(Vec,PetscInt*,PetscInt*); 367014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetOwnershipRanges(Vec,const PetscInt *[]); 3688ed539a5SBarry Smith 369014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping); 370014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 37188b03592SBarry Smith 37288b03592SBarry Smith /*MC 37388b03592SBarry Smith VecSetValueLocal - Set a single entry into a vector using the local numbering 37488b03592SBarry Smith 37588b03592SBarry Smith Synopsis: 376aaa7dc30SBarry Smith #include <petscvec.h> 3778122439eSJed Brown PetscErrorCode VecSetValueLocal(Vec v,PetscInt row,PetscScalar value, InsertMode mode); 37888b03592SBarry Smith 37988b03592SBarry Smith Not Collective 38088b03592SBarry Smith 38188b03592SBarry Smith Input Parameters: 38288b03592SBarry Smith + v - the vector 38388b03592SBarry Smith . row - the row location of the entry 38488b03592SBarry Smith . value - the value to insert 38588b03592SBarry Smith - mode - either INSERT_VALUES or ADD_VALUES 38688b03592SBarry Smith 38788b03592SBarry Smith Notes: 38888b03592SBarry Smith For efficiency one should use VecSetValues() and set several or 38988b03592SBarry Smith many values simultaneously if possible. 39088b03592SBarry Smith 39188b03592SBarry Smith These values may be cached, so VecAssemblyBegin() and VecAssemblyEnd() 39288b03592SBarry Smith MUST be called after all calls to VecSetValues() have been completed. 39388b03592SBarry Smith 39488b03592SBarry Smith VecSetValues() uses 0-based indices in Fortran as well as in C. 39588b03592SBarry Smith 39688b03592SBarry Smith Level: beginner 39788b03592SBarry Smith 39888b03592SBarry Smith .seealso: VecSetValues(), VecAssemblyBegin(), VecAssemblyEnd(), VecSetValuesBlockedLocal(), VecSetValue() 39988b03592SBarry Smith M*/ 40088b03592SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecSetValueLocal(Vec v,PetscInt i,PetscScalar va,InsertMode mode) {return VecSetValuesLocal(v,1,&i,&va,mode);} 40188b03592SBarry Smith 402014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesBlockedLocal(Vec,PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 403014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGetLocalToGlobalMapping(Vec,ISLocalToGlobalMapping*); 40490f02eecSBarry Smith 405014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotBegin(Vec,Vec,PetscScalar *); 406014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecDotEnd(Vec,Vec,PetscScalar *); 407014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotBegin(Vec,Vec,PetscScalar *); 408014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecTDotEnd(Vec,Vec,PetscScalar *); 409014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormBegin(Vec,NormType,PetscReal *); 410014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNormEnd(Vec,NormType,PetscReal *); 411d3c178dbSBarry Smith 412014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 413014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 414014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotBegin(Vec,PetscInt,const Vec[],PetscScalar[]); 415014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMTDotEnd(Vec,PetscInt,const Vec[],PetscScalar[]); 416014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscCommSplitReductionBegin(MPI_Comm); 417a751f32aSSatish Balay 418a751f32aSSatish Balay 419bda0d2a0SJed Brown typedef enum {VEC_IGNORE_OFF_PROC_ENTRIES,VEC_IGNORE_NEGATIVE_INDICES,VEC_SUBSET_OFF_PROC_ENTRIES} VecOption; 420014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOption(Vec,VecOption,PetscBool ); 42190f02eecSBarry Smith 4225c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArray(Vec,PetscScalar**); 4235c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecGetArrayRead(Vec,const PetscScalar**); 4245c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArray(Vec,PetscScalar**); 4255c0c2446SJed Brown PETSC_EXTERN PetscErrorCode VecRestoreArrayRead(Vec,const PetscScalar**); 426046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVector(Vec,Vec); 427046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVector(Vec,Vec); 428046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecGetLocalVectorRead(Vec,Vec); 429046dc149SDominic Meiser PETSC_EXTERN PetscErrorCode VecRestoreLocalVectorRead(Vec,Vec); 430d9ca1df4SBarry Smith /* 431d9ca1df4SBarry Smith Accesses a pair of pointers for two vectors that may be common. When not common the first is read only 432d9ca1df4SBarry Smith */ 433d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecGetArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 434d9ca1df4SBarry Smith { 435d9ca1df4SBarry Smith PetscErrorCode ierr; 436d9ca1df4SBarry Smith 437d9ca1df4SBarry Smith PetscFunctionBegin; 438d9ca1df4SBarry Smith ierr = VecGetArray(y,yv);CHKERRQ(ierr); 439d9ca1df4SBarry Smith if (x != y) { 440d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 441d9ca1df4SBarry Smith } else { 442d9ca1df4SBarry Smith *xv = *yv; 443d9ca1df4SBarry Smith } 444d9ca1df4SBarry Smith PetscFunctionReturn(0); 445d9ca1df4SBarry Smith } 446d9ca1df4SBarry Smith PETSC_STATIC_INLINE PetscErrorCode VecRestoreArrayPair(Vec x,Vec y,PetscScalar **xv,PetscScalar **yv) 447d9ca1df4SBarry Smith { 448d9ca1df4SBarry Smith PetscErrorCode ierr; 449d9ca1df4SBarry Smith 450d9ca1df4SBarry Smith PetscFunctionBegin; 451d9ca1df4SBarry Smith ierr = VecRestoreArray(y,yv);CHKERRQ(ierr); 452d9ca1df4SBarry Smith if (x != y) { 453d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,(const PetscScalar **)xv);CHKERRQ(ierr); 454d9ca1df4SBarry Smith } 455d9ca1df4SBarry Smith PetscFunctionReturn(0); 456d9ca1df4SBarry Smith } 457d9ca1df4SBarry Smith 458d9ca1df4SBarry Smith #if defined(PETSC_USE_DEBUG) 459d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockGet(Vec,PetscInt*); 460d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockPush(Vec); 461d9ca1df4SBarry Smith PETSC_EXTERN PetscErrorCode VecLockPop(Vec); 462d9ca1df4SBarry Smith #define VecLocked(x,arg) do {PetscInt _st; PetscErrorCode __ierr = VecLockGet(x,&_st); CHKERRQ(__ierr); if (_st > 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE," Vec is locked read only, argument # %d",arg);} while (0) 463d9ca1df4SBarry Smith #else 464e5a281bcSBarry Smith #define VecLockGet(x,arg) *(arg) = 0 465e5a281bcSBarry Smith #define VecLockPush(x) 0 466e5a281bcSBarry Smith #define VecLockPop(x) 0 467e5a281bcSBarry Smith #define VecLocked(x,arg) 468d9ca1df4SBarry Smith #endif 469e1fa1e0fSSatish Balay 47062796dfbSBarry Smith PETSC_EXTERN PetscErrorCode VecValidValues(Vec,PetscInt,PetscBool); 471f7ecc322SBarry Smith 47215091d37SBarry Smith /* 47315091d37SBarry Smith These numbers need to match the entries in 4743c94ec11SBarry Smith the function table in vecimpl.h 47515091d37SBarry Smith */ 47692d03f5aSMatthew G. Knepley typedef enum { VECOP_DUPLICATE = 0, VECOP_VIEW = 33, VECOP_LOAD = 41, VECOP_VIEWNATIVE = 68, VECOP_LOADNATIVE = 69 } VecOperation; 477014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecSetOperation(Vec,VecOperation,void(*)(void)); 478b19c1e4cSBarry Smith 479e182c471SBarry Smith /* 480e182c471SBarry Smith Routines for dealing with ghosted vectors: 481e182c471SBarry Smith vectors with ghost elements at the end of the array. 482e182c471SBarry Smith */ 483014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecMPISetGhost(Vec,PetscInt,const PetscInt[]); 484014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhost(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 485014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 486014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlock(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],Vec*); 487014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateGhostBlockWithArray(MPI_Comm,PetscInt,PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscScalar[],Vec*); 488014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostGetLocalForm(Vec,Vec*); 489014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostRestoreLocalForm(Vec,Vec*); 4903efe6655SBarry Smith PETSC_EXTERN PetscErrorCode VecGhostIsLocalForm(Vec,Vec,PetscBool*); 491014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateBegin(Vec,InsertMode,ScatterMode); 492014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecGhostUpdateEnd(Vec,InsertMode,ScatterMode); 493e182c471SBarry Smith 494014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecConjugate(Vec); 49534233285SBarry Smith 496014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToAll(Vec,VecScatter*,Vec*); 497014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterCreateToZero(Vec,VecScatter*,Vec*); 498bba1ac68SSatish Balay 4994473680cSBarry Smith PETSC_EXTERN PetscErrorCode ISComplementVec(IS,Vec,IS*); 5004473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecPow(Vec, PetscScalar); 5014473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecMedian(Vec, Vec, Vec, Vec); 5024473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetween(Vec, Vec, Vec, IS *); 5034473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichBetweenOrEqual(Vec, Vec, Vec, IS *); 5044473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichGreaterThan(Vec, Vec, IS * ); 5054473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichLessThan(Vec, Vec, IS *); 5064473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecWhichEqual(Vec, Vec, IS *); 5074473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISAXPY(Vec, IS, PetscScalar,Vec); 5084473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecISSet(Vec,IS, PetscScalar); 5094473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecBoundGradientProjection(Vec, Vec, Vec, Vec, Vec); 5104473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepBoundInfo(Vec,Vec,Vec,Vec,PetscReal*, PetscReal*,PetscReal*); 5114473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMax(Vec, Vec, PetscReal *); 5124473680cSBarry Smith PETSC_EXTERN PetscErrorCode VecStepMaxBounded(Vec,Vec,Vec,Vec,PetscReal*); 5134473680cSBarry Smith 514014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaGetVector(PetscViewer, Vec); 515014dd563SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerMathematicaPutVector(PetscViewer, Vec); 5167dbadf16SMatthew Knepley 517d59c15a7SBarry Smith /*S 518d59c15a7SBarry Smith Vecs - Collection of vectors where the data for the vectors is stored in 519759e7b9cSHong Zhang one contiguous memory 520d59c15a7SBarry Smith 521d59c15a7SBarry Smith Level: advanced 522d59c15a7SBarry Smith 523d59c15a7SBarry Smith Notes: 524d59c15a7SBarry Smith Temporary construct for handling multiply right hand side solves 525d59c15a7SBarry Smith 526d59c15a7SBarry Smith This is faked by storing a single vector that has enough array space for 527d59c15a7SBarry Smith n vectors 528d59c15a7SBarry Smith 529d59c15a7SBarry Smith Concepts: parallel decomposition 530d59c15a7SBarry Smith 531d59c15a7SBarry Smith S*/ 53295fbd943SSatish Balay struct _n_Vecs {PetscInt n; Vec v;}; 53395fbd943SSatish Balay typedef struct _n_Vecs* Vecs; 534e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDestroy(Vecs); 535e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeq(MPI_Comm,PetscInt,PetscInt,Vecs*); 536e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsCreateSeqWithArray(MPI_Comm,PetscInt,PetscInt,PetscScalar*,Vecs*); 537e1317eeeSBarry Smith PETSC_EXTERN PetscErrorCode VecsDuplicate(Vecs,Vecs*); 538e9fa29b7SSatish Balay 5398154be41SBarry Smith #if defined(PETSC_HAVE_CUSP) 540a5891931SBarry Smith typedef struct _p_PetscCUSPIndices* PetscCUSPIndices; 54125ec7418SKarl Rupp typedef struct _p_VecScatterCUSPIndices_StoS* VecScatterCUSPIndices_StoS; 54225ec7418SKarl Rupp typedef struct _p_VecScatterCUSPIndices_PtoP* VecScatterCUSPIndices_PtoP; 543014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyToGPUSome_Public(Vec,PetscCUSPIndices); 544014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCUSPCopyFromGPUSome_Public(Vec,PetscCUSPIndices); 545014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode); 546014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 547014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateSeqCUSP(MPI_Comm,PetscInt,Vec*); 548014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateMPICUSP(MPI_Comm,PetscInt,PetscInt,Vec*); 549*82f73ecaSAlejandro Lamas Daviña #elif defined(PETSC_HAVE_VIENNACL) 550b17c682bSKarl Rupp typedef struct _p_PetscViennaCLIndices* PetscViennaCLIndices; 551b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesCreate(PetscInt, PetscInt*,PetscInt, PetscInt*,PetscViennaCLIndices*); 552b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLIndicesDestroy(PetscViennaCLIndices*); 553b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyToGPUSome_Public(Vec,PetscViennaCLIndices); 554b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLCopyFromGPUSome_Public(Vec,PetscViennaCLIndices); 555b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCL(MPI_Comm,PetscInt,Vec*); 556b17c682bSKarl Rupp PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCL(MPI_Comm,PetscInt,PetscInt,Vec*); 557*82f73ecaSAlejandro Lamas Daviña #elif defined(PETSC_HAVE_VECCUDA) 558*82f73ecaSAlejandro Lamas Daviña typedef struct _p_PetscCUDAIndices* PetscCUDAIndices; 559*82f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_StoS* VecScatterCUDAIndices_StoS; 560*82f73ecaSAlejandro Lamas Daviña typedef struct _p_VecScatterCUDAIndices_PtoP* VecScatterCUDAIndices_PtoP; 561*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCUDACopyToGPUSome_Public(Vec,PetscCUDAIndices); 562*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCUDACopyFromGPUSome_Public(Vec,PetscCUDAIndices); 563*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecScatterInitializeForGPU(VecScatter,Vec,ScatterMode); 564*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecScatterFinalizeForGPU(VecScatter); 565*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateSeqCUDA(MPI_Comm,PetscInt,Vec*); 566*82f73ecaSAlejandro Lamas Daviña PETSC_EXTERN PetscErrorCode VecCreateMPICUDA(MPI_Comm,PetscInt,PetscInt,Vec*); 567b17c682bSKarl Rupp #endif 568b17c682bSKarl Rupp 569014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVecs(Vec,PetscInt*,Vec**); 570014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSubVec(Vec,PetscInt,Vec*); 571014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVecs(Vec,PetscInt,PetscInt*,Vec*); 572014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestSetSubVec(Vec,PetscInt,Vec); 573014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecCreateNest(MPI_Comm,PetscInt,IS*,Vec*,Vec*); 574014dd563SJed Brown PETSC_EXTERN PetscErrorCode VecNestGetSize(Vec,PetscInt*); 57545b63f25SDmitry Karpeev 576c5929fdfSBarry Smith PETSC_EXTERN PetscErrorCode PetscOptionsGetVec(PetscOptions,const char[],const char[],Vec,PetscBool*); 5774325cce7SMatthew G Knepley PETSC_EXTERN PetscErrorCode VecChop(Vec,PetscReal); 578ab352700SBarry Smith 579077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecGetLayout(Vec,PetscLayout*); 580077aedafSJed Brown PETSC_EXTERN PetscErrorCode VecSetLayout(Vec,PetscLayout); 5819a6d0b0bSJed Brown 5829a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode PetscSectionVecView(PetscSection, Vec, PetscViewer); 5839a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecGetValuesSection(Vec, PetscSection, PetscInt, PetscScalar **); 5849a6d0b0bSJed Brown PETSC_EXTERN PetscErrorCode VecSetValuesSection(Vec, PetscSection, PetscInt, PetscScalar [], InsertMode); 58522d96a0cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscSectionVecNorm(PetscSection, PetscSection, Vec, NormType, PetscReal []); 5869a6d0b0bSJed Brown 587ad59f098SAta Mesgarnejad PETSC_EXTERN PetscErrorCode PetscSFCreateFromZero(MPI_Comm,Vec,PetscSF*); 588ad59f098SAta Mesgarnejad 5892eac72dbSBarry Smith #endif 5902b82984cSSatish Balay 591