From c353b0522a2eb7e3a2a4f07de5b3bfdb30a99f49 Mon Sep 17 00:00:00 2001 From: Matthew Poremba Date: Wed, 30 Mar 2022 12:20:26 -0500 Subject: [PATCH] configs: Add option for mem type for GPUFS The user will likely want to be able to specify a different type of memory for the GPU rather than using the same default for the CPU. Add the option to do that. Change-Id: I62c22f6283335c3ca3df355f8ecc4bbffa751a73 Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/58390 Reviewed-by: Jason Lowe-Power Maintainer: Jason Lowe-Power Tested-by: kokoro --- configs/example/gpufs/runfs.py | 4 +++- configs/ruby/GPU_VIPER.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/configs/example/gpufs/runfs.py b/configs/example/gpufs/runfs.py index fae985a561..b198552b71 100644 --- a/configs/example/gpufs/runfs.py +++ b/configs/example/gpufs/runfs.py @@ -82,7 +82,9 @@ def addRunFSOptions(parser): " size") parser.add_argument("--dgpu-num-dirs", type=int, default=1, help="Set " "the number of dGPU directories (memory controllers") - + parser.add_argument("--dgpu-mem-type", default="HBM_1000_4H_1x128", + choices=ObjectList.mem_list.get_names(), + help="type of memory to use") def runGpuFSSystem(args): ''' diff --git a/configs/ruby/GPU_VIPER.py b/configs/ruby/GPU_VIPER.py index c73995f6c0..dc99429b47 100644 --- a/configs/ruby/GPU_VIPER.py +++ b/configs/ruby/GPU_VIPER.py @@ -518,7 +518,7 @@ def construct_gpudirs(options, system, ruby_system, network): dir_cntrl.responseFromMemory = MessageBuffer() # Create memory controllers too - mem_type = ObjectList.mem_list.get(options.mem_type) + mem_type = ObjectList.mem_list.get(options.dgpu_mem_type) dram_intf = MemConfig.create_mem_intf(mem_type, gpu_mem_range, i, int(math.log(options.dgpu_num_dirs, 2)), options.cacheline_size, xor_low_bit)