Lines Matching refs:chunk_slice_map
22 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
666 chunk_slice_map = hipstruct->chunk_slice_map; in MatMult_SeqSELLHIP()
668 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMult_SeqSELLHIP()
670 …, dim3(WARP_SIZE, 4)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMult_SeqSELLHIP()
672 …, dim3(WARP_SIZE, 8)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMult_SeqSELLHIP()
674 … dim3(WARP_SIZE, 16)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMult_SeqSELLHIP()
676 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMult_SeqSELLHIP()
720 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMultAdd_SeqSELLHIP() local
758 chunk_slice_map = hipstruct->chunk_slice_map; in MatMultAdd_SeqSELLHIP()
760 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
762 …, dim3(WARP_SIZE, 4)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
764 …, dim3(WARP_SIZE, 8)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
766 … dim3(WARP_SIZE, 16)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
768 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
820 chunk_slice_map = hipstruct->chunk_slice_map; in MatMultAdd_SeqSELLHIP()
822 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
824 …, dim3(WARP_SIZE, 4)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
826 …, dim3(WARP_SIZE, 8)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
828 … dim3(WARP_SIZE, 16)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()
830 …, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map, acolidx, ava… in MatMultAdd_SeqSELLHIP()