SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

enable more test cases on xpu (#38572)

Browse Source 
* enable glm4 integration cases on XPU, set xpu expectation for blip2

Signed-off-by: Matrix YAO <matrix.yao@intel.com>

* more

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix style

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* refine wording

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* refine test case names

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* run

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* add gemma2 and chameleon

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix review comments

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

---------

Signed-off-by: Matrix YAO <matrix.yao@intel.com>
Signed-off-by: YAO Matrix <matrix.yao@intel.com>
This commit is contained in:
Yao Matrix
2025-06-06 15:29:51 +08:00
committed by GitHub
parent 31023b6909
commit 89542fb81c
23 changed files with 150 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/models/blip_2/test_modeling_blip_2.py
View File

@@ -1708,10 +1708,14 @@ class Blip2ModelIntegrationTest(unittest.TestCase):
expectations = Expectations(
{
("xpu", 3): [
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
"a woman is playing with her dog on the beach",
],
("cuda", 7): [
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
"a woman is playing with her dog on the beach",
]
],
}
)
expected_outputs = expectations.get_expectation()
@@ -1729,10 +1733,14 @@ class Blip2ModelIntegrationTest(unittest.TestCase):
expectations = Expectations(
{
("xpu", 3): [
[0, 3, 7, 152, 2515, 11389, 3523, 1],
"san francisco",
],
("cuda", 7): [
[0, 3, 7, 152, 2515, 11389, 3523, 1],
"san francisco",
]
],
}
)
expected_outputs = expectations.get_expectation()
@@ -1755,10 +1763,14 @@ class Blip2ModelIntegrationTest(unittest.TestCase):
expectations = Expectations(
{
("xpu", 3): [
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
],
("cuda", 7): [
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
[0, 3, 9, 2335, 19, 1556, 28, 160, 1782, 30, 8, 2608, 1],
]
],
}
)
expected_predictions = expectations.get_expectation()