Skip to content

Commit 038d24c

Browse files
authored
[dgx.b200] add runner type (#697)
1 parent 627f73b commit 038d24c

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.ci/test_infra/oss_ci_benchmark_v3.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,11 @@
2424
"gpu_count": 1,
2525
"avail_gpu_mem_in_gb": 288,
2626
},
27+
"linux.dgx.b200": {
28+
"name": "linux.dgx.b200",
29+
"gpu_count": 1,
30+
"avail_gpu_mem_in_gb": 192,
31+
}
2732
}
2833

2934

@@ -52,7 +57,7 @@ def setup_tritonbench_cwd():
5257
def parse_runners(
5358
runner_name: str, runner_type: str, envs: Dict[str, str]
5459
) -> List[Dict[str, Any]]:
55-
runner_mapping = RUNNER_TYPE_MAPPING[runner_type].copy()
60+
runner_mapping = RUNNER_TYPE_MAPPING.get(runner_type, {}).copy()
5661
runner_mapping["name"] = runner_name
5762
runner_mapping["gpu_info"] = envs["device"]
5863
runner_mapping["extra_info"] = {}

0 commit comments

Comments
 (0)