Skip to content

Commit d6fc50b

Browse files
committed
feat(gaudi/ci): added ci for gaudi device
1 parent 76632c8 commit d6fc50b

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

integration-tests/fixtures/gaudi/service.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -190,27 +190,28 @@ def get_free_port():
190190
except Exception as e:
191191
logger.error(f"Error handling existing container: {str(e)}")
192192

193-
model_name = next(
194-
name for name, cfg in TEST_CONFIGS.items() if cfg["model_id"] == model_id
195-
)
196-
197-
tgi_args = TEST_CONFIGS[model_name]["args"].copy()
193+
tgi_args = TEST_CONFIGS[test_name]["args"].copy()
198194

199195
env = BASE_ENV.copy()
200196

201197
# Add model_id to env
202198
env["MODEL_ID"] = model_id
203199

204200
# Add env config that is definied in the fixture parameter
205-
if "env_config" in TEST_CONFIGS[model_name]:
206-
env.update(TEST_CONFIGS[model_name]["env_config"].copy())
201+
if "env_config" in TEST_CONFIGS[test_name]:
202+
env.update(TEST_CONFIGS[test_name]["env_config"].copy())
207203

208204
volumes = [f"{DOCKER_VOLUME}:/data"]
209205
logger.debug(f"Using volume {volumes}")
210206

211207
try:
208+
logger.debug(f"Using command {tgi_args}")
212209
logger.info(f"Creating container with name {container_name}")
213210

211+
logger.debug(f"Using environment {env}")
212+
logger.debug(f"Using volumes {volumes}")
213+
logger.debug(f"HABANA_RUN_ARGS {HABANA_RUN_ARGS}")
214+
214215
# Log equivalent docker run command for debugging, this is not actually executed
215216
container = client.containers.run(
216217
DOCKER_IMAGE,

integration-tests/gaudi/capture_expected_outputs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from typing import Dict, Any, Generator
44

55
import pytest
6-
from test_generate import TEST_CONFIGS
6+
from test_gaudi_generate import TEST_CONFIGS
77

88
UNKNOWN_CONFIGS = {
99
name: config

0 commit comments

Comments
 (0)