Use batch_size fixture

archana-ramalingam · archana-ramalingam · commit 03bbcedf9c9a · 2024-11-22T04:00:46.000Z
diff --git a/sharktank/tests/evaluate/perplexity_vmfb_test.py b/sharktank/tests/evaluate/perplexity_vmfb_test.py
@@ -27,6 +27,7 @@
     "get_iree_flags",
     "tensor_parallelism_size",
     "baseline_perplexity_scores",
+    "batch_size",
 )
 @is_mi300x
 class PerplexityTest(unittest.TestCase):

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,7 @@`
`27`	`27`	`"get_iree_flags",`
`28`	`28`	`"tensor_parallelism_size",`
`29`	`29`	`"baseline_perplexity_scores",`
	`30`	`+ "batch_size",`
`30`	`31`	`)`
`31`	`32`	`@is_mi300x`
`32`	`33`	`class PerplexityTest(unittest.TestCase):`