-
Notifications
You must be signed in to change notification settings - Fork 28.2k
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
ENH: [CI
] Add new workflow to run slow tests of important models on push main if they are modified
#29235
Conversation
run: | ||
pytest -m "flash_attn_test" tests/${{ matrix.model-name }}/test_modeling_* | ||
pytest -m "integration_test" tests/${{ matrix.model-name }}/test_modeling_* | ||
- name: Post results Slack channel |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this can be simplified using a re-usable workflow
@@ -529,6 +529,7 @@ def test_flash_attn_2_equivalence(self): | |||
|
|||
@require_torch_gpu | |||
@slow | |||
@pytest.mark.integration_test |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
let's update the pytest toml for all IntegrationTest
if possible
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nice!
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
…-new-push-test-workflow
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
…ngface/transformers into add-new-push-test-workflow
The workflow is ready ! to add whisper / llava it will require some work, I 'll do that in a follow up PR! |
id: run_fa2_tests | ||
run: | ||
pytest -m "flash_attn_test" --make-reports=${{ matrix.model-name }}_fa2_tests/ tests/${{ matrix.model-name }}/test_modeling_* | ||
- name: "Test suite reports artifacts: ${{ matrix.model-name }}_fa2_tests" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
let's separate each step by an empty line - easier to read 🙏
title: 🤗 Results of the Integration tests - ${{ matrix.model-name }} | ||
status: ${{ steps.run_integration_tests.conclusion}} | ||
slack_token: ${{ secrets.CI_SLACK_BOT_TOKEN }} | ||
- name: Tailscale |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
add a comment to say what is this for
Left 2 nits, otherwise LGTM. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks Younes, let's get it on main!
As per title, WIP ..
cc @ydshieh