Lines Matching defs:sizeBytes

240 mgpuMemAlloc(uint64_t sizeBytes, CUstream stream, bool isHostShared) {
243   if (sizeBytes == 0)
248         cuMemAllocManaged(&ptr, sizeBytes, CU_MEM_ATTACH_GLOBAL));
251   CUDA_REPORT_IF_ERROR(cuMemAlloc(&ptr, sizeBytes));
261 mgpuMemcpy(void *dst, void *src, size_t sizeBytes, CUstream stream) {
264                                      sizeBytes, stream));
286 mgpuMemHostRegister(void *ptr, uint64_t sizeBytes) {
288   CUDA_REPORT_IF_ERROR(cuMemHostRegister(ptr, sizeBytes, /*flags=*/0));
308   uint64_t sizeBytes = sizes[0] * denseStrides[0] * elementSizeBytes;
316   mgpuMemHostRegister(ptr, sizeBytes);