Home
last modified time | relevance | path

Searched refs:vecs_gpu (Results 1 – 4 of 4) sorted by relevance

/petsc/src/ts/impls/explicit/rk/
H A Drk.c973 …ssianProductFunctionPU(ts, rk->stage_time, Y[i], VecsSensiTemp, ts->vec_sensip_col, ts->vecs_gpu)); in TSAdjointStep_RK()
978 …RHSHessianProductFunctionPU(quadts, rk->stage_time, Y[i], NULL, ts->vec_sensip_col, ts->vecs_gpu)); in TSAdjointStep_RK()
1009 PetscCall(VecAXPY(VecDeltaMu2, -h * b[i], ts->vecs_gpu[nadj])); in TSAdjointStep_RK()
1013 PetscCall(VecAXPY(VecDeltaMu2, -h, ts->vecs_gpu[nadj])); in TSAdjointStep_RK()
/petsc/include/petsc/private/
H A Dtsimpl.h229 Vec *vecs_fpu, *vecs_gpu; member
/petsc/src/ts/interface/sensitivity/
H A Dtssen.c709 if (rhshp3) ts->vecs_gpu = rhshp3; in TSSetRHSHessianProduct()
/petsc/src/ts/impls/implicit/theta/
H A Dtheta.c1077 if (!ts->ihessianproduct_fpu) ts->vecs_fpu = ts->vecs_gpu; in TSAdjointSetUp_Theta()