Skip to content

Commit 1eed8ba

Browse files
authored
[CI][BugFix] Adapt test_token_penalty to Paddle uniform float16 behavior change (#8031)
1 parent bbe5f81 commit 1eed8ba

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

tests/cov_pytest.ini

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,3 @@ addopts =
1414
--ignore=tests/graph_optimization/test_cuda_graph_dynamic_subgraph.py
1515
--ignore=tests/e2e/test_ernie_03b_pd_decode_unified_attention.py
1616
--ignore=tests/e2e/test_ernie_03b_pd_router_v1_ipc.py
17-
--ignore=tests/operators/test_token_penalty.py

tests/operators/test_token_penalty.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,10 @@ def test_token_penalty_once(self):
4444
for orig, penal in zip(original_values.numpy(), penalized_values.numpy()):
4545
if orig < 0:
4646
self.assertLess(penal, orig, "负值应该乘以惩罚因子")
47-
else:
47+
elif orig > 0:
4848
self.assertLess(penal, orig, "正值应该除以惩罚因子")
49+
else:
50+
self.assertEqual(penal, orig, "零值不受惩罚影响")
4951

5052
def test_compare_with_naive_implementation(self):
5153
res = get_token_penalty_once(self.pre_ids, self.logits, self.penalty_scores)

0 commit comments

Comments
 (0)