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)