Skip to content

Skip test_multi_gpu_data_parallel_forward for MobileViTV2ModelTest#24017

Merged
ydshieh merged 3 commits into
mainfrom
skip_a_test_in_mobilevitv2
Jun 5, 2023
Merged

Skip test_multi_gpu_data_parallel_forward for MobileViTV2ModelTest#24017
ydshieh merged 3 commits into
mainfrom
skip_a_test_in_mobilevitv2

Conversation

@ydshieh

@ydshieh ydshieh commented Jun 5, 2023

Copy link
Copy Markdown
Collaborator

What does this PR do?

Skip test_multi_gpu_data_parallel_forward for MobileViTV2ModelTest.

This passes on CI, but the other 2x tests running after this one are all failing with CUDA error: misaligned address.

(Similar to #21991)

@ydshieh ydshieh requested a review from amyeroberts June 5, 2023 12:36
@HuggingFaceDocBuilderDev

HuggingFaceDocBuilderDev commented Jun 5, 2023

Copy link
Copy Markdown

The documentation is not available anymore as the PR was closed or merged.

@amyeroberts amyeroberts left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Thanks for fixing!

Do you know if this happens with any other models? It would be good to keep track in case there's something in the architecture design we need to be careful about.

@ydshieh

ydshieh commented Jun 5, 2023

Copy link
Copy Markdown
Collaborator Author

Yes, see #21991. That PR mentioned torch 2.0, but this test are skipped for some other modes even before that PR for other reasons.

@ydshieh ydshieh merged commit 5176dc2 into main Jun 5, 2023
@ydshieh ydshieh deleted the skip_a_test_in_mobilevitv2 branch June 5, 2023 14:29
novice03 pushed a commit to novice03/transformers that referenced this pull request Jun 23, 2023
huggingface#24017)

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
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.

3 participants