8000 [ci] add test workflow for 3090 machine by zhulinJulia24 · Pull Request #3561 · InternLM/lmdeploy · GitHub
[go: up one dir, main page]

Skip to content

[ci] add test workflow for 3090 machine #3561

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 30 commits into
base: main
Choose a base branch
from

Conversation

zhulinJulia24
Copy link
Collaborator

add test workflow for 3090 machine

@zhulinJulia24 zhulinJulia24 requested a review from Copilot May 15, 2025 08:13
Copy link
@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR adds support for running tests on a 3090 machine by introducing new configuration flags, updating test annotations, and adding a dedicated workflow file. The key changes include:

  • Injecting environment‑specific command line flags (e.g. “cache‑max‑entry‑count”) for the 3090 machine.
  • Updating test functions to include a new @pytest.mark.test_3090 annotation.
  • Adding a new daily ETE test workflow file and a dedicated 3090 configuration file.

Reviewed Changes

Copilot reviewed 21 out of 21 changed files in this pull request and generated 2 comments.

Show a summary per file
File Description
autotest/utils/run_client_chat.py Adds extra CLI flag when running on env '3090'.
autotest/utils/quantization_utils.py Adjusts batch-size logic for 3090 and applies conditional cuda_prefix usage.
aut 8000 otest/utils/pipeline_chat.py Updates default parameter handling for extra and injects cache settings for 3090.
autotest/utils/config_utils.py Uses OrderedDict to remove duplicate model entries in quantization config.
Various test files Adds @pytest.mark.test_3090 annotations for tests targeting the 3090 machine.
autotest/tools/pipeline/mllm_case.py & llm_case.py Adds support for passing cache settings via extra configuration.
autotest/config.yaml & config-3090.yaml Introduces env_tag for distinguishing configurations.
.github/workflows/daily_ete_test_3090.yml New workflow configuration for daily testing on the 3090 machine.

@@ -14,7 +14,7 @@ def run_pipeline_chat_test(config,
model_case,
backend_type,
worker_id: str = '',
extra: object = None,
extra: object = {},
Copy link
Preview
Copilot AI May 15, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Avoid using a mutable default argument ({}); consider using None as the default and initializing extra inside the function to prevent unintended shared state across calls.

Suggested change
extra: object = {},
extra: object = None,

Copilot uses AI. Check for mistakes.

@zhulinJulia24 zhulinJulia24 requested a review from Copilot May 16, 2025 02:28
Copy link
@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR adds support for running test workflows on a 3090 machine by updating various test files, configuration files, and adding a new GitHub workflow. Key changes include:

  • Inserting environment‐specific logic (checking for env_tag "3090") in the client, quantization, and pipeline modules.
  • Adding new pytest marks (@pytest.mark.test_3090) to multiple test cases and updating configuration files to support 3090 settings.
  • Introducing a new GitHub workflow file (.github/workflows/daily_ete_test_3090.yml) to orchestrate daily tests on the 3090 machine.

Reviewed Changes

Copilot reviewed 23 out of 23 changed files in this pull request and generated 2 comments.

Show a summary per file
File Description
autotest/utils/run_client_chat.py Added environment check for "3090" to adjust extra cli flags.
autotest/utils/quantization_utils.py Refactored command construction for quantization with environment-specific batch sizes.
autotest/utils/pipeline_chat.py Included test-specific overrides via env_tag "3090" and modified extra parameters for chat tests.
autotest/utils/config_utils.py Updated quantization model list construction to remove duplicate entries using OrderedDict.
Various test files under autotest/tools Added new pytest marker (@pytest.mark.test_3090) for 3090-targeted tests.
autotest/config.yaml & autotest/config-3090.yaml Introduced and updated env_tag settings to distinguish between standard and 3090 environments.
.github/workflows/daily_ete_test_3090.yml Added a comprehensive workflow to run daily E2E tests on the 3090 machine.
Comments suppressed due to low confidence (1)

autotest/utils/pipeline_chat.py:33

  • [nitpick] The extra parameter 'cache-max-entry-count' is set to different values across modules (e.g., 0.7 in run_client_chat, 0.6 and 0.5 in pipeline tests). Verify these values are intentionally different; if not, consider consolidating them to avoid confusion.
if str(config.get('env_tag')) == '3090':

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant
0