Skip to content

Commit 5b2105a

Browse files
rohansjoshifacebook-github-bot
authored andcommitted
Re-enable eval script (pytorch#14176)
Summary: Revised diff to enable eval_llama_qnn script which does evaluations with prefill model Will add unit test in a later diff Reviewed By: cccclai Differential Revision: D82143474
1 parent 5d63bad commit 5b2105a

File tree

2 files changed

+212
-84
lines changed

2 files changed

+212
-84
lines changed

examples/qualcomm/oss_scripts/llama/TARGETS

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ python_binary(
9090
"//executorch/examples/qualcomm/oss_scripts/llama:range_setting_pt2e",
9191
"fbsource//third-party/pypi/lm-eval:lm-eval",
9292
],
93+
keep_gpu_sections = True,
9394
)
9495

9596
runtime.command_alias(

0 commit comments

Comments
 (0)