Skip to content

Commit

Permalink
#7525: Increase BERT on-device perf threshold, we got a bump I guess
Browse files Browse the repository at this point in the history
  • Loading branch information
tt-rkim committed Mar 7, 2025
1 parent 8455e8a commit a156677
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def test_perf_device_bare_metal(batch_size, test, expected_perf):
"batch_size, test, expected_perf",
[
[7, "BERT_LARGE-batch_8-BFLOAT8_B-SHARDED", 280],
[8, "BERT_LARGE-batch_8-BFLOAT8_B-SHARDED", 340],
[8, "BERT_LARGE-batch_8-BFLOAT8_B-SHARDED", 360],
],
)
def test_perf_device_bare_metal_wh(batch_size, test, expected_perf):
Expand Down

0 comments on commit a156677

Please sign in to comment.