Searched refs:mrctx (Results 1 – 2 of 2) sorted by relevance
321 Status XRTMemoryManager::TryFreeMemoryStep(MemoryReclaimContext* mrctx, in TryFreeMemoryStep() argument323 DeviceContext* device_context = GetDeviceContext(mrctx->device_ordinal, in TryFreeMemoryStep()328 if (!mrctx->done_freeing) { in TryFreeMemoryStep()334 (mrctx->requested_free_size > 0) in TryFreeMemoryStep()335 ? std::min<size_t>(mrctx->requested_free_size - mrctx->free_size, in TryFreeMemoryStep()340 device_context->TryFreeMemory(mrctx->backend, free_size); in TryFreeMemoryStep()345 mrctx->free_size += size; in TryFreeMemoryStep()350 mrctx->done_freeing = true; in TryFreeMemoryStep()352 if (!mrctx->done_compacting) { in TryFreeMemoryStep()353 mrctx->done_compacting = true; in TryFreeMemoryStep()[all …]
151 Status TryFreeMemoryStep(MemoryReclaimContext* mrctx, const Status& status);161 MemoryReclaimContext mrctx(backend, device_ordinal, requested_free_size); in Run()171 TF_RETURN_IF_ERROR(TryFreeMemoryStep(&mrctx, result_or.status())); in Run()