minor fix for tests/examples/llm_ptq/test_llm_ptq.py

Edwardf0t1 · web-flow · commit 070063c1beb3 · 2025-09-17T09:55:06.000-07:00
Signed-off-by: Zhiyu &lt;bestczy317@gmail.com&gt;
diff --git a/tests/examples/llm_ptq/test_llm_ptq.py b/tests/examples/llm_ptq/test_llm_ptq.py
@@ -110,7 +110,7 @@ def test_ptq_whisper(self, command):
         PTQCommand(quant="nvfp4", min_sm=100),
         #
         # multi_gpu
-        PTQCommand(quant="nvfp4", min_gpu=2),
+        PTQCommand(quant="nvfp4", min_gpu=2, tp=2),
     ],
     ids=PTQCommand.param_str,
 )

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def test_ptq_whisper(self, command):`
`110`	`110`	`PTQCommand(quant="nvfp4", min_sm=100),`
`111`	`111`	`#`
`112`	`112`	`# multi_gpu`
`113`		`- PTQCommand(quant="nvfp4", min_gpu=2),`
	`113`	`+ PTQCommand(quant="nvfp4", min_gpu=2, tp=2),`
`114`	`114`	`],`
`115`	`115`	`ids=PTQCommand.param_str,`
`116`	`116`	`)`