Skip to content

Commit 5415acb

Browse files
committed
variable update
1 parent ca36c16 commit 5415acb

File tree

1 file changed

+2
-1
lines changed
  • examples/qualcomm/oss_scripts/llama

1 file changed

+2
-1
lines changed

examples/qualcomm/oss_scripts/llama/llama.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -843,6 +843,7 @@ def post_process():
843843
)
844844

845845
runner_cmd = ""
846+
performance_output_path = "outputs/inference_speed.txt"
846847
if args.enable_x86_64:
847848
# x86 emulator is intended for CI and not performance. Check only the first few tokens.
848849
seq_len = min(seq_len, 16)
@@ -862,6 +863,7 @@ def post_process():
862863
f"--model_path {pte_path}",
863864
f"--seq_len {seq_len}",
864865
f"--output_path {args.artifact}/outputs/outputs.txt",
866+
f"--performance_output_path {performance_output_path}",
865867
f"--kv_updater ShiftPointer",
866868
runner_args,
867869
]
@@ -874,7 +876,6 @@ def post_process():
874876
)
875877
post_process()
876878
else:
877-
performance_output_path = "outputs/inference_speed.txt"
878879
runner_cmd = " ".join(
879880
[
880881
f"cd {workspace} &&",

0 commit comments

Comments
 (0)