Home
last modified time | relevance | path

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

/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu19 PetscInt *chunk_slice_map; /* starting slice for each chunk, device pointer */ member
29 if ((*cudastruct)->chunk_slice_map) PetscCallCUDA(cudaFree((*cudastruct)->chunk_slice_map)); in MatSeqSELLCUDA_Destroy()
51 if (cudastruct->chunk_slice_map) PetscCallCUDA(cudaFree(cudastruct->chunk_slice_map)); in MatSeqSELLCUDACopyToGPU()
64 …CallCUDA(cudaMalloc((void **)&cudastruct->chunk_slice_map, a->totalchunks * sizeof(*cudastruct->ch… in MatSeqSELLCUDACopyToGPU()
65 …etscCallCUDA(cudaMemcpy(cudastruct->chunk_slice_map, a->chunk_slice_map, a->totalchunks * sizeof(*… in MatSeqSELLCUDACopyToGPU()
185 …ight, PetscInt chunksperblock, PetscInt totalchunks, const PetscInt *chunk_slice_map, const PetscI… in matmult_seqsell_tiled_kernel8() argument
199 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmult_seqsell_tiled_kernel8()
240 …ight, PetscInt chunksperblock, PetscInt totalchunks, const PetscInt *chunk_slice_map, const PetscI… in matmultadd_seqsell_tiled_kernel8() argument
254 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmultadd_seqsell_tiled_kernel8()
579 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMult_SeqSELLCUDA() local
[all …]
/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx22 PetscInt *chunk_slice_map; /* starting slice for each chunk, device pointer */ member
32 if ((*hipstruct)->chunk_slice_map) PetscCallHIP(hipFree((*hipstruct)->chunk_slice_map)); in MatSeqSELLHIP_Destroy()
54 if (hipstruct->chunk_slice_map) PetscCallHIP(hipFree(hipstruct->chunk_slice_map)); in MatSeqSELLHIPCopyToGPU()
67 …tscCallHIP(hipMalloc((void **)&hipstruct->chunk_slice_map, a->totalchunks * sizeof(*hipstruct->chu… in MatSeqSELLHIPCopyToGPU()
68 …PetscCallHIP(hipMemcpy(hipstruct->chunk_slice_map, a->chunk_slice_map, a->totalchunks * sizeof(*a-… in MatSeqSELLHIPCopyToGPU()
189 …ight, PetscInt chunksperblock, PetscInt totalchunks, const PetscInt *chunk_slice_map, const PetscI… in matmult_seqsell_tiled_kernel8() argument
203 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmult_seqsell_tiled_kernel8()
244 …ight, PetscInt chunksperblock, PetscInt totalchunks, const PetscInt *chunk_slice_map, const PetscI… in matmultadd_seqsell_tiled_kernel8() argument
258 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmultadd_seqsell_tiled_kernel8()
584 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMult_SeqSELLHIP() local
[all …]
/petsc/src/mat/impls/sell/seq/
H A Dsell.h58 PetscInt *chunk_slice_map; /* starting slice of the current chunk, CUDA only */ \
H A Dsell.c906 PetscCall(PetscFree(a->chunk_slice_map)); in MatDestroy_SeqSELL()
1496 PetscCall(PetscFree(a->chunk_slice_map)); in MatAssemblyEnd_SeqSELL()
1497 PetscCall(PetscMalloc1(totalchunks, &a->chunk_slice_map)); in MatAssemblyEnd_SeqSELL()
1503 a->chunk_slice_map[i] = j; in MatAssemblyEnd_SeqSELL()