Home
last modified time | relevance | path

Searched refs:PetscCallCXX (Results 1 – 25 of 36) sorted by relevance

12

/petsc/src/mat/tests/
H A Dex302k.kokkos.cxx83 PetscCallCXX(a_local_d = Kokkos::View<PetscScalar *>("a_local_d", di[5])); in main()
84 PetscCallCXX(i_local_d = Kokkos::View<PetscInt *>("i_local_d", m + 1)); in main()
85 PetscCallCXX(j_local_d = Kokkos::View<PetscInt *>("j_local_d", di[5])); in main()
88 PetscCallCXX(a_nonlocal_d = Kokkos::View<PetscScalar *>("a_nonlocal_d", oi[5])); in main()
89 PetscCallCXX(i_nonlocal_d = Kokkos::View<PetscInt *>("i_nonlocal_d", m + 1)); in main()
90 PetscCallCXX(j_nonlocal_d = Kokkos::View<PetscInt *>("j_nonlocal_d", oi[5])); in main()
103 PetscCallCXX(a_local_h = Kokkos::View<PetscScalar *, HostMirrorMemorySpace>(da, di[5])); in main()
104 PetscCallCXX(i_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(di, m + 1)); in main()
105 PetscCallCXX(j_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(dj, di[5])); in main()
106 PetscCallCXX(a_nonlocal_h = Kokkos::View<PetscScalar *, HostMirrorMemorySpace>(oa, oi[5])); in main()
[all …]
/petsc/src/sys/tests/
H A Dex64.cxx59 PetscCallCXX(buf = std::to_string(x) + ", " + std::to_string(y)); in to_string()
89 PetscCallCXX(buf = '<'); in to_string()
91 PetscCallCXX(buf += std::to_string(x[i])); in to_string()
92 if (i + 1 != x.size()) PetscCallCXX(buf += ", "); in to_string()
94 PetscCallCXX(buf += ">, <" + y + '>'); in to_string()
180 PetscCallCXX(oss << std::boolalpha); in view_map()
181 PetscCallCXX(oss << "map: '" << this->map_name << "'\n"); in view_map()
182 PetscCallCXX(oss << " size: " << map.size() << '\n'); in view_map()
183 PetscCallCXX(oss << " capacity: " << map.capacity() << '\n'); in view_map()
184 PetscCallCXX(oss << " bucket count: " << map.bucket_count() << '\n'); in view_map()
[all …]
/petsc/src/sys/objects/kokkos/
H A Dkinit.kokkos.cxx16 PetscCallCXX(delete PetscKokkosExecutionSpacePtr); in PetscKokkosFinalize_Private()
18 PetscCallCXX(Kokkos::kokkos_free(PetscScalarPool)); in PetscKokkosFinalize_Private()
21 PetscCallCXX(Kokkos::finalize()); in PetscKokkosFinalize_Private()
72 PetscCallCXX(Kokkos::initialize(args)); in PetscKokkosInitializeCheck()
85 …if (dtype == PETSC_DEVICE_CUDA) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExe… in PetscKokkosInitializeCheck()
87 …if (dtype == PETSC_DEVICE_HIP) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExec… in PetscKokkosInitializeCheck()
92 …PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(*(sycl::queue *)hand… in PetscKokkosInitializeCheck()
97 PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace()); in PetscKokkosInitializeCheck()
103 …PetscCallCXX(PetscScalarPool = static_cast<PetscScalar *>(Kokkos::kokkos_malloc(sizeof(PetscScalar… in PetscKokkosInitializeCheck()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp74 PetscCallCXX(v_dual = PetscScalarKokkosDualView(v_d, v_h)); in UpdateArray()
84 PetscCallCXX(v_dual = PetscScalarKokkosDualView(v_dual.view<DefaultMemorySpace>(), v_h)); in UpdateArray()
85 PetscCallCXX(v_dual.modify_host()); in UpdateArray()
95 PetscCallCXX(v_dual = PetscScalarKokkosDualView(v_d, v_dual.view_host())); in UpdateArray()
96 PetscCallCXX(v_dual.modify_device()); in UpdateArray()
103 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
104 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
111 …PetscCallCXX(jmap1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
112 …PetscCallCXX(perm1_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
113 …PetscCallCXX(imap2_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosV… in SetUpCOO()
[all …]
H A Dveckok.kokkos.cxx26 PetscCallCXX(veckok->v_dual.sync<ExecutionSpace>()); in VecGetKokkosView_Private()
28 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView_Private()
40 …if (overwrite) PetscCallCXX(veckok->v_dual.clear_sync_state()); /* If overwrite=true, clear the ol… in VecRestoreKokkosView_Private()
41 PetscCallCXX(veckok->v_dual.modify<ExecutionSpace>()); in VecRestoreKokkosView_Private()
54 PetscCallCXX(veckok->v_dual.sync<ExecutionSpace>()); in VecGetKokkosView()
55 PetscCallCXX(*kv = veckok->v_dual.view<ExecutionSpace>()); in VecGetKokkosView()
127 PetscCallCXX(KokkosBlas::abs(exec, xv, xv)); in VecAbs_SeqKokkos()
141 PetscCallCXX(Kokkos::parallel_for( in VecReciprocal_SeqKokkos()
158 PetscCallCXX(Kokkos::parallel_reduce( in VecMin_SeqKokkos()
182 PetscCallCXX(Kokkos::parallel_reduce( in VecMax_SeqKokkos()
[all …]
/petsc/src/sys/objects/device/tests/
H A Dex1k.kokkos.cxx41 PetscCallCXX(exec.fence()); // Initialize device runtime to get more accurate timing below in main()
44 …for (i = 0; i < n; i++) PetscCallCXX(Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const PetscInt &i)… in main()
46 PetscCallCXX(exec.fence()); in main()
53 PetscCallCXX(Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const PetscInt &i){})); in main()
54 PetscCallCXX(exec.fence()); in main()
H A Dex11.cxx83 PetscCallCXX(buf.fill(0)); in CheckMarkedObjectMap_Private()
104 PetscCallCXX(cont_ids.reserve(cont.size())); in TestAllCombinations()
109 PetscCallCXX(cont_ids.emplace_back(id)); in TestAllCombinations()
127 PetscCallCXX(found_keys.resize(nkeys)); in TestAllCombinations()
234 PetscCallCXX(expected_map.erase(found_it)); in CheckMapEqual()
/petsc/src/vec/is/sf/tests/
H A Dex4k.kokkos.cxx76 PetscCallCXX(rootdata = (PetscScalar *)Kokkos::kokkos_malloc(sizeof(PetscScalar) * maxn)); in main()
77 PetscCallCXX(leafdata = (PetscScalar *)Kokkos::kokkos_malloc(sizeof(PetscScalar) * maxn)); in main()
121 PetscCallCXX(Kokkos::deep_copy(dst1, src1)); in main()
122 PetscCallCXX(Kokkos::deep_copy(dst2, src2)); in main()
133 PetscCallCXX(Kokkos::fence()); in main()
142 PetscCallCXX(Kokkos::fence()); in main()
157 PetscCallCXX(Kokkos::kokkos_free(rootdata)); in main()
158 PetscCallCXX(Kokkos::kokkos_free(leafdata)); in main()
/petsc/src/sys/objects/device/impls/sycl/
H A Dsyclcontext.sycl.cxx101 if (!idle) PetscCallCXX(dci->queue.wait()); in synchronize()
136 PetscCallCXX(dci->timeBegin = std::chrono::steady_clock::now()); in beginTimer()
150 PetscCallCXX(dci->queue.wait()); in endTimer()
151 PetscCallCXX(duration = std::chrono::steady_clock::now() - dci->timeBegin); in endTimer()
152 PetscCallCXX(*elapsed = duration.count()); in endTimer()
177 PetscCallCXX(dctx->data = new DeviceContext::PetscDeviceContext_SYCL()); in PetscDeviceContextCreate_SYCL()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx277 PetscCallCXX(Kokkos::deep_copy(Ti_h, 0)); in MatSeqAIJKokkosGenerateTransposeStructure()
303 …PetscCallCXX(T_d = KokkosCsrMatrix("csrmatT", n, m, nz, MatScalarKokkosView("Ta", nz), Ti_d, Tj_d)… in MatSeqAIJKokkosGenerateTransposeStructure()
304 PetscCallCXX(perm_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), perm_h)); in MatSeqAIJKokkosGenerateTransposeStructure()
332 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
339 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateTranspose_Private()
370 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateHermitian_Private()
377 …PetscCallCXX(Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, nz), KO… in MatSeqAIJKokkosGenerateHermitian_Private()
398 …PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), "N", 1.0 /*alpha*/, aijkok->csrmat… in MatMult_SeqAIJKokkos()
429 …PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0… in MatMultTranspose_SeqAIJKokkos()
459 …PetscCallCXX(KokkosSparse::spmv(PetscGetKokkosExecutionSpace(), mode, 1.0 /*alpha*/, csrmat, xv, 0… in MatMultHermitianTranspose_SeqAIJKokkos()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx177 PetscCallCXX(delete *reinterpret_cast<MatProductCtx_MPIAIJKokkos **>(data)); in MatProductCtxDestroy_MPIAIJKokkos()
617 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d)); in MatMPIAIJKokkosReduceBegin()
618 …PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); // Fo's column si… in MatMPIAIJKokkosReduceBegin()
646 PetscCallCXX(Kokkos::parallel_for( in MatMPIAIJKokkosReduceBegin()
692 PetscCallCXX(Kokkos::parallel_for( in MatMPIAIJKokkosReduceEnd()
699 PetscCallCXX(Kokkos::parallel_for( in MatMPIAIJKokkosReduceEnd()
930 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d)); in MatMPIAIJKokkosBcastBegin()
931 PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); in MatMPIAIJKokkosBcastBegin()
973 PetscCallCXX(Kokkos::parallel_for( in MatMPIAIJKokkosBcastBegin()
1024 PetscCallCXX(Kokkos::parallel_for( in MatMPIAIJKokkosBcastEnd()
[all …]
/petsc/src/sys/objects/device/interface/
H A Dpetscdevice_interface_internal.hpp122 PetscCallCXX(self_.reset(dctx, NoOpDeleter{})); in reset_self()
124 PetscCallCXX(self_.reset()); in reset_self()
132 PetscCallCXX(this->upstream().clear()); in clear()
133 PetscCallCXX(this->marked_objects().clear()); in clear()
146 PetscCallCXX(marked_objects_.emplace(id)); in add_mark()
150 if (!pair.first) PetscCallCXX(marked_objects_.insert(pair.second, id)); in add_mark()
H A Ddcontext.cxx21 PetscCallCXX(PetscObjectCast(dctx)->cpp = new CxxData{dctx}); in construct_()
532 PetscCallCXX(CxxDataCast(dctxa)->upstream()[bid] = CxxDataCast(dctxb)->weak_snapshot()); in PetscDeviceContextWaitForContext()
599 if (PetscDefined(USE_DEBUG_AND_INFO)) PetscCallCXX(idList.reserve(4 * n)); in PetscDeviceContextForkWithStreamType()
637 PetscCallCXX(idList += std::to_string(childID)); in PetscDeviceContextForkWithStreamType()
638 if (ninput != 1) PetscCallCXX(idList += ", "); in PetscDeviceContextForkWithStreamType()
766 if (PetscDefined(USE_DEBUG_AND_INFO)) PetscCallCXX(idList.reserve(4 * n)); in PetscDeviceContextJoin()
773 PetscCallCXX(idList += std::to_string(PetscObjectCast((*dsub)[i])->id)); in PetscDeviceContextJoin()
774 if (i + 1 < n) PetscCallCXX(idList += ", "); in PetscDeviceContextJoin()
875 PetscCallCXX(dvec.clear()); in PetscDeviceContextGetNullContextForDevice_Private()
890 if (static_cast<std::size_t>(devid) >= ctxlist.size()) PetscCallCXX(ctxlist.resize(devid + 1)); in PetscDeviceContextGetNullContextForDevice_Private()
[all …]
H A Dmemory.cxx135 PetscCallCXX(map = map_type{}); in finalize_()
213 PetscCallCXX(map.emplace( in PetscDeviceRegisterMemory_Private()
426 PetscCallCXX(map.erase(found_it)); in PetscDeviceDeallocate_Private()
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx60 PetscCallCXX(bs_dual.modify_host()); in BuildHelperArrays()
61 PetscCallCXX(bs2_dual.modify_host()); in BuildHelperArrays()
62 PetscCallCXX(blkMap_dual.modify_host()); in BuildHelperArrays()
90 PetscCallCXX(Kokkos::parallel_for( in PCApplyOrTranspose_VPBJacobi_Kokkos()
107 PetscCallCXX(Kokkos::parallel_for( in PCApplyOrTranspose_VPBJacobi_Kokkos()
139 PetscCallCXX(delete static_cast<PC_VPBJacobi_Kokkos *>(jac->spptr)); in PCDestroy_VPBJacobi_Kokkos()
172 PetscCallCXX(delete pckok); in PCSetUp_VPBJacobi_Kokkos()
178 PetscCallCXX(pckok = new PC_VPBJacobi_Kokkos(nlocal, nblocks, nsize, bsizes)); in PCSetUp_VPBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/pbjacobi/kokkos/
H A Dpbjacobi_kok.kokkos.cxx21 PetscCallCXX(diag_dual.modify_host()); /* mark the host has newer data */ in Update()
47 PetscCallCXX(Kokkos::parallel_for( in PCApplyOrTranspose_PBJacobi_Kokkos()
77 PetscCallCXX(delete static_cast<PC_PBJacobi_Kokkos *>(jac->spptr)); in PCDestroy_PBJacobi_Kokkos()
91 PetscCallCXX(jac->spptr = new PC_PBJacobi_Kokkos(len, const_cast<PetscScalar *>(jac->diag))); in PCSetUp_PBJacobi_Kokkos()
/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmevent.hpp41 PetscCallCXX(pool_.pop()); in finalize_()
55 PetscCallCXX(*event = std::move(pool_.top())); in allocate()
56 PetscCallCXX(pool_.pop()); in allocate()
68 PetscCallCXX(pool_.push(std::move(event))); in deallocate()
/petsc/src/sys/objects/cxx/memory/
H A Dobject_pool.cxx135 PetscCallCXX(::delete[] reinterpret_cast<unsigned char *>(header)); in delete_ptr_()
191 PetscCallCXX(this->pool().clear()); in clear_()
330 …PetscCallCXX(base_ptr = reinterpret_cast<unsigned char *>(util::construct_at(reinterpret_cast<Allo… in allocate_ptr_()
447 PetscCallCXX(ptr_list.pop_back()); in try_allocate()
515 …if (it == this->pool().end() || it->first != align) PetscCallCXX(it = this->pool().insert(it, {ali… in deallocate()
516 PetscCallCXX(it->second[size].emplace_back(ptr)); in deallocate()
/petsc/include/petsc/private/
H A Dkokkosimpl.hpp136 PetscCallCXX(v_dual.sync_host(exec)); in KokkosDualViewSyncHost()
142 PetscCallCXX(exec.fence()); in KokkosDualViewSyncHost()
153 PetscCallCXX(v_dual.sync_device(exec)); in KokkosDualViewSyncDevice()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx107 …PetscCallCXX(w_dual = PetscScalarKokkosDualView("w_dual", win->map->n + wdata->nghost)); // Kokkos… in VecDuplicate_MPIKokkos()
278 PetscCallCXX(v->spptr = new Vec_Kokkos(v->map->n, vecmpi->array, NULL)); in VecConvert_MPI_MPIKokkos_inplace()
302 …PetscCallCXX(w_dual = PetscScalarKokkosDualView("VecDuplicateVecs", m * lda)); // Kokkos init's w_… in VecDuplicateVecs_MPIKokkos_GEMV()
310 …PetscCallCXX(static_cast<Vec_Kokkos *>(v->spptr)->v_dual.modify_host()); // as we only init'ed arr… in VecDuplicateVecs_MPIKokkos_GEMV()
354 …PetscCallCXX(v_dual = PetscScalarKokkosDualView("v_dual", v->map->n)); // Kokkos init's v_dual to … in VecCreate_MPIKokkos()
360 PetscCallCXX(v->spptr = new Vec_Kokkos(v_dual)); in VecCreate_MPIKokkos()
391 …PetscCallCXX(w->spptr = new Vec_Kokkos(map->n, const_cast<PetscScalar *>(harray), const_cast<Petsc… in VecCreateMPIKokkosWithLayoutAndArrays_Private()
503 …PetscCallCXX(w->spptr = new Vec_Kokkos(n, const_cast<PetscScalar *>(harray), const_cast<PetscScala… in VecCreateMPIKokkosWithArrays_Private()
/petsc/src/ksp/pc/impls/vpbjacobi/cuda/
H A Dvpbjacobi_cuda.cu155 PetscCallCXX(delete static_cast<PC_VPBJacobi_CUDA *>(jac->spptr)); in PCDestroy_VPBJacobi_CUDA()
175 PetscCallCXX(delete pcuda); in PCSetUp_VPBJacobi_CUDA()
180 PetscCallCXX(jac->spptr = new PC_VPBJacobi_CUDA(n, nblocks, nsize, bsizes, jac->diag)); in PCSetUp_VPBJacobi_CUDA()
/petsc/src/vec/vec/tests/
H A Dex16k.kokkos.cxx23 PetscCallCXX(Kokkos::deep_copy(kv, 2.0)); in main()
/petsc/include/petsc/private/cpp/
H A Dobject_pool.hpp166 PetscCallCXX(it = decltype(it){ptr_stack.erase(std::next(it).base())}); in for_each()
286 PetscCallCXX(util::construct_at(ptr, std::forward<Args>(args)...)); in construct_()
295 PetscCallCXX(util::destroy_at(ptr)); in destroy_()
H A Dunordered_map.hpp695 PetscCallCXX(values_[it] = internal_value_type{}); in khash_erase_()
943 PetscCallCXX(flags_.clear()); in resize()
966 if (old_n_buckets < new_n_buckets) PetscCallCXX(values_.resize(new_n_buckets)); in resize()
992 PetscCallCXX(swap(values_[j], key)); in resize()
998 if (i != j) PetscCallCXX(values_[j] = std::move(key)); in resize()
1005 if (old_n_buckets > new_n_buckets) PetscCallCXX(values_.resize(new_n_buckets)); in resize()
1006 PetscCallCXX(flags_ = std::move(new_flags)); in resize()
1017 PetscCallCXX(values_.clear()); in clear()
1018 PetscCallCXX(std::fill(flags_.begin(), flags_.end(), default_bit_pattern())); in clear()
1343 PetscCallCXX(this->erase(it)); in erase()
/petsc/src/sys/objects/device/impls/
H A Dsegmentedmempool.hpp355 PetscCallCXX(chunks_.clear()); in clear_()
441 PetscCallCXX(chunks_.emplace_back(block_alloced, req_size)); in try_allocate_chunk()
476 PetscCallCXX(chunks_.pop_back()); in try_allocate_chunk()
729 PetscCallCXX(pool_.emplace_back(&allocator_, block_size, stream)); in make_block_()
746 PetscCallCXX(pool_.clear()); in finalize_()

12