Searched refs:pytorch_memory_limits (Results 1 – 1 of 1) sorted by relevance
100 std::vector<size_t> pytorch_memory_limits; variable174 pytorch_memory_limits[device] = UINT64_MAX; in lazy_init_device()354 if (pytorch_used_bytes[device] + size > pytorch_memory_limits[device]) { in mallocAsync()385 format_size(pytorch_memory_limits[device])); in mallocAsync()438 pytorch_memory_limits.resize(dev_count); in init()472 pytorch_memory_limits[device] = in setMemoryFraction()543 pytorch_memory_limits[device] - pytorch_used_bytes[device]); in cacheInfo()555 if (pytorch_used_bytes[device] + guess > pytorch_memory_limits[device]) { in cacheInfo()