Skip to content

Commit 05fe088

Browse files
committed
fix format error
Signed-off-by: Jinzhen Lin <[email protected]>
1 parent 1458451 commit 05fe088

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

tests/kernels/test_marlin_gemm.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -231,13 +231,12 @@ def test_gptq_marlin_gemm(
231231
workspace = MarlinWorkspace(size_n, GPTQ_MARLIN_MIN_THREAD_N,
232232
GPTQ_MARLIN_MAX_PARALLEL)
233233

234-
opcheck(
235-
torch.ops._C.gptq_marlin_gemm,
236-
(a_input, marlin_q_w, marlin_s, marlin_zp, g_idx, sort_indices,
237-
workspace.scratch, quant_type.id, a_input.shape[0], b_weight.shape[1],
238-
a_input.shape[1], is_k_full, False, use_atomic_add, use_fp32_reduce,
239-
False),
240-
test_utils=DEFAULT_OPCHECK_TEST_UTILS)
234+
opcheck(torch.ops._C.gptq_marlin_gemm,
235+
(a_input, marlin_q_w, marlin_s, marlin_zp, g_idx, sort_indices,
236+
workspace.scratch, quant_type.id, a_input.shape[0],
237+
b_weight.shape[1], a_input.shape[1], is_k_full, False,
238+
use_atomic_add, use_fp32_reduce, False),
239+
test_utils=DEFAULT_OPCHECK_TEST_UTILS)
241240

242241
output = ops.gptq_marlin_gemm(
243242
a_input,

0 commit comments

Comments
 (0)