Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
benchmarks/pyproject.toml
View file @ e9db304c
...@@ -40,7 +40,7 @@ classifiers = [ ...@@ -40,7 +40,7 @@ classifiers = [
] ]
dependencies = [ 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", "aiperf==0.6.0",
"matplotlib", "matplotlib",
"networkx", "networkx",
......
components/src/dynamo/profiler/thorough.py
View file @ e9db304c
...@@ -356,6 +356,7 @@ async def run_thorough( ...@@ -356,6 +356,7 @@ async def run_thorough(
isl=isl, isl=isl,
osl=osl, osl=osl,
num_gpus_per_node=dgdr.hardware.numGpusPerNode, num_gpus_per_node=dgdr.hardware.numGpusPerNode,
total_gpus=total_gpus,
k8s_pvc_name=model_cache.pvcName, k8s_pvc_name=model_cache.pvcName,
k8s_pvc_mount_path=model_cache.pvcMountPath, k8s_pvc_mount_path=model_cache.pvcMountPath,
k8s_model_path_in_pvc=model_cache.pvcModelPath, k8s_model_path_in_pvc=model_cache.pvcModelPath,
......
container/deps/requirements.planner.txt
View file @ e9db304c
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
# Dependencies required by the planner, profiler, and global_planner services. # 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 aiofiles<=25.1.0
aiohttp>=3.9.0,<4.0 aiohttp>=3.9.0,<4.0
filterpy==1.4.5 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