-
Notifications
You must be signed in to change notification settings - Fork 28.8k
enable internvl UTs on XPU #37779
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
base: main
Are you sure you want to change the base?
enable internvl UTs on XPU #37779
Conversation
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Hi 👋, thank you for opening this pull request! The pull request is converted to draft by default. The CI will be paused while the PR is in draft mode. When it is ready for review, please click the |
@ydshieh , pls help review, thx |
@@ -3887,8 +3885,8 @@ def test_sdpa_can_compile_dynamic(self): | |||
self.skipTest(reason="This test requires an NVIDIA GPU with compute capability >= 8.0") | |||
elif device_type == "rocm" and major < 9: | |||
self.skipTest(reason="This test requires an AMD GPU with compute capability >= 9.0") | |||
else: | |||
self.skipTest(reason="This test requires a Nvidia or AMD GPU") | |||
elif device_type not in ["cuda", "rocm", "xpu"]: |
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.
there's no device type "rocm"
("xpu", 3): 'user\n\nWhat are the differences between these two images?\nassistant\nThe images show the Statue of Liberty and the Golden Gate Bridge from different angles. Here are the differences:\n\n1. **Foreground', | ||
("cuda", 7): 'user\n\nWhat are the differences between these two images?\nassistant\nThe images show the Statue of Liberty and the Golden Gate Bridge from different angles. Here are the differences:\n\n1. **Angle', | ||
} | ||
) # fmt: skip |
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.
why is # fmt: skip
needed here ?
pytest -rA tests/models/internvl/test_modeling_internvl.py
137 cases PASSED, 4 failed cases are same as CUDA, will investigate in separate efforts.
FAILED tests/models/internvl/test_modeling_internvl.py::InternVLModelTest::test_resize_embeddings_untied_with_deepspeed
FAILED tests/models/internvl/test_modeling_internvl.py::InternVLModelTest::test_resize_embeddings_untied_with_deepspeed_multi_gpu
FAILED tests/models/internvl/test_modeling_internvl.py::InternVLModelTest::test_resize_tokens_embeddings_with_deepspeed
FAILED tests/models/internvl/test_modeling_internvl.py::InternVLModelTest::test_resize_tokens_embeddings_with_deepspeed_multi_gpu