diff --git a/.buildkite/test-amd.yaml b/.buildkite/test-amd.yaml index ef30158430..f3ac37b0e8 100644 --- a/.buildkite/test-amd.yaml +++ b/.buildkite/test-amd.yaml @@ -720,7 +720,7 @@ steps: #-------------------------------------------------------- mi300 ยท detokenizer --------------------------------------------------------# - label: ":amd: (MI300) Async Engine, Inputs, Utils, Worker" # TBD - timeout_in_minutes: 35 + timeout_in_minutes: 50 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] dind: false agent_pool: mi300_1 @@ -878,7 +878,7 @@ steps: - pytest -v -s distributed/test_elastic_ep.py - label: ":amd: (MI300) RayExecutorV2" # TBD - timeout_in_minutes: 60 + timeout_in_minutes: 75 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] dind: false agent_pool: mi300_4 @@ -1074,7 +1074,7 @@ steps: - pytest -v -s entrypoints/openai/completion --ignore=entrypoints/openai/completion/test_tensorizer_entrypoint.py - label: ":amd: (MI300) Entrypoints Integration (API Server Generate)" # TBD - timeout_in_minutes: 50 + timeout_in_minutes: 65 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] dind: false agent_pool: mi300_1 @@ -2163,7 +2163,7 @@ steps: - pytest -v -s models/multimodal/processing/test_tensor_schema.py - label: ":amd: (MI300) Multimodal Models (Extended Pooling)" # TBD - timeout_in_minutes: 50 + timeout_in_minutes: 75 mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300] dind: false agent_pool: mi300_1 diff --git a/.buildkite/test_areas/models_multimodal.yaml b/.buildkite/test_areas/models_multimodal.yaml index 1207e674e5..9893383bff 100644 --- a/.buildkite/test_areas/models_multimodal.yaml +++ b/.buildkite/test_areas/models_multimodal.yaml @@ -239,7 +239,7 @@ steps: label: ":amd: (MI300) Multimodal Models (Extended Pooling)" dind: false device: mi300_1 - timeout_in_minutes: 60 + timeout_in_minutes: 75 depends_on: - image-build-amd source_file_dependencies: