Searched refs:vmem_ (Results 1 – 2 of 2) sorted by relevance
120 vmem_(vmem), in GpuVirtualMemAllocator()128 GpuDriver::FreeVirtualMemory(&gpu_context_, vmem_); in ~GpuVirtualMemAllocator()136 GpuDevicePtr next_va = vmem_.base + next_alloc_offset_; in Alloc()141 if (next_va + padded_bytes > vmem_.base + vmem_.size_bytes) { in Alloc()211 next_alloc_offset_ = mapping_it->va - vmem_.base; in Free()
91 stream_executor::gpu::GpuDriver::VmemSpan vmem_; variable