Skip to content

Commit

Permalink
#0: Increase thresholds for device perf: falcon7b prefill, resnet lof…
Browse files Browse the repository at this point in the history
…i batch 16
  • Loading branch information
tt-rkim committed May 10, 2024
1 parent 1086d8c commit 54881fd
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion models/demos/resnet/tests/test_perf_device_resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
[
[16, "HiFi2-activations_BFLOAT8_B-weights_BFLOAT8_B-batch_16-24576", 5420],
[20, "HiFi2-activations_BFLOAT8_B-weights_BFLOAT8_B-batch_20-24576", 5780],
[16, "LoFi-activations_BFLOAT8_B-weights_BFLOAT8_B-batch_16-24576", 6730],
[16, "LoFi-activations_BFLOAT8_B-weights_BFLOAT8_B-batch_16-24576", 6940],
[20, "LoFi-activations_BFLOAT8_B-weights_BFLOAT8_B-batch_20-24576", 7500],
],
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
@pytest.mark.parametrize(
"batch_size, test, expected_perf",
[
[1, "BFLOAT16-L1-falcon_7b-layers_32-prefill_seq256", 3.27],
[1, "BFLOAT16-L1-falcon_7b-layers_32-prefill_seq256", 3.49],
[32, "BFLOAT16-L1-falcon_7b-layers_32-decode_batch32", 139],
],
)
Expand Down

0 comments on commit 54881fd

Please sign in to comment.