From 64d5b905410ad2e2eb5b38d8f5703ef50f825cf8 Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Wed, 24 Jul 2024 15:35:19 -0700 Subject: [PATCH] Set specific gpu 0 as well --- python/test/test_api.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/python/test/test_api.py b/python/test/test_api.py index 452429ed9..c15847aab 100644 --- a/python/test/test_api.py +++ b/python/test/test_api.py @@ -357,8 +357,11 @@ def test_stop(self): { "backend": "python", "parameters": {"decoupled": {"string_value": "False"}}, - # Keep instance count low for fast startup/cleanup - "instance_group": [{"kind": "KIND_GPU", "count": 1}], + # Keep instance count low for fast startup/cleanup. + # Alternatively can use KIND_CPU here, but keeping gpus/count explicit. + "instance_group": [ + {"kind": "KIND_GPU", "gpus": [0], "count": 1} + ], } ) },