From 1763a82a5ebc002d47edb5bc53106f9b312069b9 Mon Sep 17 00:00:00 2001 From: Avinash Sharma Date: Thu, 5 Dec 2024 14:07:31 -0800 Subject: [PATCH] Use correct args for input len 2048 llama test (#649) Uses args for input len 2048 llama test instead of args for input len 128 llama test. --------- Signed-off-by: aviator19941 --- sharktank/tests/models/llama/benchmark_amdgpu_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sharktank/tests/models/llama/benchmark_amdgpu_test.py b/sharktank/tests/models/llama/benchmark_amdgpu_test.py index 166b0b78a..13a2c35e4 100644 --- a/sharktank/tests/models/llama/benchmark_amdgpu_test.py +++ b/sharktank/tests/models/llama/benchmark_amdgpu_test.py @@ -270,7 +270,7 @@ def testBenchmark8B_f16_Non_Decomposed_Prefill_Input_Len_2048(self): hip_device_id=self.iree_device, vmfb_name=output_vmfb, irpa_path=self.irpa_path, - args=self.iree_run_prefill_nondecomposed_args_fp16, + args=self.iree_run_prefill_nondecomposed_args_fp16_2048, cwd=self.repo_root, )