Forwarding flavor parameters in stress test

[bottlenecks.git] / utils / infra_setup / runner / yardstick.py

diff --git a/utils/infra_setup/runner/yardstick.py b/utils/infra_setup/runner/yardstick.py
index 562e918..6071585 100644 (file)
--- a/utils/infra_setup/runner/yardstick.py
+++ b/utils/infra_setup/runner/yardstick.py
@@ -46,9 +46,12 @@ def yardstick_command_parser(debug, cidr, outfile, parameter):
     image_name = config.bottlenecks_config["yardstick_image_name"]
     parameter["image_name"] = image_name
     DEPLOY_SCENARIO = os.getenv("DEPLOY_SCENARIO")
+    RAM_NUM = os.getenv("RAM_NUM")
     if DEPLOY_SCENARIO:
         if "ovs" in DEPLOY_SCENARIO:
             parameter["dpdk_enabled"] = True
+            if RAM_NUM:
+                parameter["ram_num"] = RAM_NUM
     LOG.info(parameter)
     if parameter is not None:
         cmd += " --task-args " + '"' + str(parameter) + '"'