Searched refs:max_split_size (Results 1 – 6 of 6) sorted by relevance
108 int64_t max_split_size = 0; member
18 static size_t max_split_size() { in max_split_size() function
1091 stats.max_split_size = in DeviceCachingAllocator()1092 static_cast<int64_t>(CUDAAllocatorConfig::max_split_size()); in DeviceCachingAllocator()1430 if (block->size >= CUDAAllocatorConfig::max_split_size()) in alloc_found_block()1480 if (block->size >= CUDAAllocatorConfig::max_split_size()) in free()2463 return (size < CUDAAllocatorConfig::max_split_size()) && in should_split()2525 if ((p.size() < CUDAAllocatorConfig::max_split_size()) && in get_free_block()2526 ((*it)->size >= CUDAAllocatorConfig::max_split_size())) in get_free_block()2529 if ((p.size() >= CUDAAllocatorConfig::max_split_size()) && in get_free_block()2701 if (size >= CUDAAllocatorConfig::max_split_size()) in alloc_block()2729 if (CUDAAllocatorConfig::max_split_size() == in release_available_cached_blocks()[all …]
156 size_t max_split_size; member
375 max_split_size_s, int64_t(snapshot.config_metadata.max_split_size)); in _memory_snapshot_pickled()
599 result["max_split_size"] = stats.max_split_size; in THCPModule_memoryStats()825 int64_t(snapshot.config_metadata.max_split_size); in THCPModule_memorySnapshot()