Skip to content
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

Define single_gpu test marker #1201

Merged
merged 4 commits into from
Jan 22, 2025
Merged

Define single_gpu test marker #1201

merged 4 commits into from
Jan 22, 2025

Conversation

nikg4
Copy link
Collaborator

@nikg4 nikg4 commented Jan 21, 2025

Description

-- Define single_gpu test marker to allow skipping such tests on multi-GPU machines if needed.

Related issues

Towards OPE-698, OPE-909

Before submitting

  • This PR only changes documentation. (You can ignore the following checks in that case)
  • Did you read the contributor guideline Pull Request guidelines?
  • Did you link the issue(s) related to this PR in the section above?
  • Did you add / update tests where needed?

Reviewers

At least one review from a member of oumi-ai/oumi-staff is required.

Copy link

linear bot commented Jan 21, 2025

OPE-698

OPE-909

@nikg4 nikg4 requested review from optas, oelachqar and taenin January 21, 2025 20:22
@nikg4 nikg4 marked this pull request as ready for review January 21, 2025 20:22
@nikg4 nikg4 requested a review from wizeng23 January 21, 2025 20:23
@nikg4 nikg4 changed the title Update @requires_gpu test marker Update @requires_gpu test decorator Jan 21, 2025
@nikg4 nikg4 changed the title Update @requires_gpu test decorator Define single_gpu test marker Jan 21, 2025
@@ -235,4 +235,5 @@ addopts = ["--import-mode=prepend"]
markers = [
"e2e: Slow e2e integration tests",
"e2e_eternal: Extremely slow e2e integration tests (for manual/selective runs)",
"single_gpu: The test uses max 1 GPU (can be potentially skipped on multi-GPU machine to conserve GPU resources)",
Copy link
Contributor

Choose a reason for hiding this comment

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

Given most tests run with 1 gpu, maybe we should mark multi-gpu instead? e.g. multigpu_160gb

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Ack. Will iterate on these.

@nikg4 nikg4 merged commit ddbaaaf into main Jan 22, 2025
2 checks passed
@nikg4 nikg4 deleted the xrdaukar/test-max-gpu-count branch January 22, 2025 04:38
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.

2 participants