Unverified Commit e9db304c authored by Hongkuan Zhou's avatar Hongkuan Zhou Committed by GitHub
Browse files

fix(profiler): pass total_gpus to AIC enumerate so candidates respect budget (#8479)


Signed-off-by: default avatarhongkuanz <hongkuanz@nvidia.com>
parent d250fab9
......@@ -40,7 +40,7 @@ classifiers = [
]
dependencies = [
"aiconfigurator[webapp] @ git+https://github.com/ai-dynamo/aiconfigurator.git@1fe8084c817194827706947cc175107ee9e6ea25",
"aiconfigurator[webapp] @ git+https://github.com/ai-dynamo/aiconfigurator.git@b33a136e78f9351a7ad17db1e2a1abd7b295cc6a",
"aiperf==0.6.0",
"matplotlib",
"networkx",
......
......@@ -356,6 +356,7 @@ async def run_thorough(
isl=isl,
osl=osl,
num_gpus_per_node=dgdr.hardware.numGpusPerNode,
total_gpus=total_gpus,
k8s_pvc_name=model_cache.pvcName,
k8s_pvc_mount_path=model_cache.pvcMountPath,
k8s_model_path_in_pvc=model_cache.pvcModelPath,
......
......@@ -3,7 +3,7 @@
# Dependencies required by the planner, profiler, and global_planner services.
aiconfigurator[webapp] @ git+https://github.com/ai-dynamo/aiconfigurator.git@1fe8084c817194827706947cc175107ee9e6ea25
aiconfigurator[webapp] @ git+https://github.com/ai-dynamo/aiconfigurator.git@b33a136e78f9351a7ad17db1e2a1abd7b295cc6a
aiofiles<=25.1.0
aiohttp>=3.9.0,<4.0
filterpy==1.4.5
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment