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

fix XPU UT error case brough by RNG difference btw XPU and CUDA (#37121)

Browse Source 
* fix XPU UT error case brough by RNG difference btw XPU and CUDA

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

* enable tests/models/llama/test_modeling_llama.py::LlamaIntegrationTest::test_model_7b_logits and tests/models/llama/test_modeling_llama.py::LlamaIntegrationTest::test_model_7b_logits_bf16 on xpu

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

* Revert "enable tests/models/llama/test_modeling_llama.py::LlamaIntegrationTest::test_model_7b_logits and tests/models/llama/test_modeling_llama.py::LlamaIntegrationTest::test_model_7b_logits_bf16 on xpu"

This reverts commit 3ef83a4f0204642daa45fda56e8aca1afed24b4f.

---------

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
This commit is contained in:
Yao Matrix
2025-04-01 20:52:55 +08:00
committed by GitHub
parent 897ff9af0e
commit 24e311f42b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/generation/test_logits_process.py
View File

@@ -976,7 +976,8 @@ class LogitsProcessorTest(unittest.TestCase):
input_ids[:, -1] = 10 input_ids[:, -1] = 10
scores_wo_bias = scores[:, -1].clone() scores_wo_bias = scores[:, -1].clone()
out = watermark(input_ids=input_ids, scores=scores) out = watermark(input_ids=input_ids, scores=scores)
self.assertTrue((out[:, 1] == scores_wo_bias + watermark.bias).all()) greenlist_id = 3 if torch_device == "xpu" else 1
self.assertTrue((out[:, greenlist_id] == scores_wo_bias + watermark.bias).all())
@parameterized.expand([(5, 3, 10000), (10, 5, 1000)]) @parameterized.expand([(5, 3, 10000), (10, 5, 1000)])
def test_synthidtext_watermarking_processor_bias_uniformity(self, ngram_len, num_layers, vocab_size): def test_synthidtext_watermarking_processor_bias_uniformity(self, ngram_len, num_layers, vocab_size):