File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -2524,13 +2524,9 @@ static vk_device ggml_vk_get_device(size_t idx) {
2524
2524
2525
2525
if (GGML_VK_SUBALLOCATION_BLOCK_SIZE != nullptr ) {
2526
2526
device->suballocation_block_size = std::stoul (GGML_VK_SUBALLOCATION_BLOCK_SIZE);
2527
- #if defined(_WIN32)
2528
- } else if (device->vendor_id == VK_VENDOR_ID_NVIDIA) {
2527
+ } else {
2529
2528
// Limit batching of allocations to 1GB by default to avoid fragmentation issues
2530
2529
device->suballocation_block_size = 1024 *1024 *1024 ;
2531
- #endif
2532
- } else {
2533
- device->suballocation_block_size = device->max_memory_allocation_size ;
2534
2530
}
2535
2531
device->suballocation_block_size = std::min (device->suballocation_block_size , device->max_memory_allocation_size );
2536
2532
You can’t perform that action at this time.
0 commit comments