Просмотр исходного кода

Fix showing GPU size in starpu_machine_display

Samuel Thibault лет назад: 12
Родитель
Сommit
d71b88efa7
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      src/drivers/cuda/driver_cuda.c

+ 2 - 0
src/drivers/cuda/driver_cuda.c

@@ -78,6 +78,8 @@ static void _starpu_cuda_limit_gpu_mem_if_needed(unsigned devid)
 	size_t STARPU_ATTRIBUTE_UNUSED to_waste = 0;
 	char name[30];
 
+	global_mem[devid] = props[devid].totalGlobalMem;
+
 	limit = starpu_get_env_number("STARPU_LIMIT_CUDA_MEM");
 	if (limit == -1)
 	{