Browse Source

Fix showing GPU size in starpu_machine_display

Samuel Thibault 12 years ago
parent
commit
d71b88efa7
1 changed files with 2 additions and 0 deletions
  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)
 	{