瀏覽代碼

Fix showing GPU size in starpu_machine_display

Samuel Thibault 12 年之前
父節點
當前提交
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)
 	{