diff --git a/configs/xpu/df_clsf.json b/configs/xpu/df_clsf.json
index 4d14763b8..0c7c25d70 100644
--- a/configs/xpu/df_clsf.json
+++ b/configs/xpu/df_clsf.json
@@ -26,6 +26,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 10,
             "max-depth": 5
         },
@@ -46,6 +47,7 @@
                     }
                 }
             ],
+            "workload-size": "large",
             "num-trees": 100,
             "max-depth": 8
         },
@@ -66,6 +68,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 20,
             "max-depth": 16
         },
@@ -86,6 +89,7 @@
                     }
                 }
             ],
+            "workload-size": "large",
             "num-trees": 100,
             "max-depth": 10
         },
@@ -106,6 +110,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 50,
             "max-depth": 15
         }
diff --git a/runner.py b/runner.py
index 980e40b87..f030a76b5 100755
--- a/runner.py
+++ b/runner.py
@@ -58,6 +58,11 @@ def get_configs(path: Path) -> List[str]:
                         help='Available floating point data types'
                         'This parameter only marks dtype as available, '
                         'make sure to add the dtype parameter to the config file ')
+    parser.add_argument('--workload-size', type=str, default="small medium large", nargs='+',
+                        choices=("small", "medium", "large"),
+                        help='Available workload sizes,'
+                        'make sure to add the workload-size parameter to the config file '
+                        'unmarked workloads will be launched anyway')
     parser.add_argument('--no-intel-optimized', default=False, action='store_true',
                         help='Use Scikit-learn without Intel optimizations')
     parser.add_argument('--output-file', default='results.json',
@@ -105,6 +110,11 @@ def get_configs(path: Path) -> List[str]:
             params = common_params.copy()
             params.update(params_set.copy())
 
+            if 'workload-size' in params:
+                if params['workload-size'] not in args.workload_size:
+                    continue
+                del params['workload-size']
+
             device = []
             if 'device' not in params:
                 if 'sklearn' in params['lib']: