Skip to content

Commit b878508

Browse files
authored
bugfix: Fix the "not enough values to unpack" error. (#540)
This PR fixes the following error: ``` File "/home/yuxianq/flashinfer/python/flashinfer/decode.py", line 616, in plan self._cached_module = get_batch_decode_module( File "/home/yuxianq/flashinfer/python/flashinfer/decode.py", line 114, in get_batch_decode_module _batch_decode_modules[args] = compile_batch_decode_module(*args) File "/home/yuxianq/flashinfer/python/flashinfer/decode.py", line 65, in compile_batch_decode_module uri, path = gen_batch_decode_cu(*args), ValueError: not enough values to unpack (expected 2, got 1) ```
1 parent d0b0b7f commit b878508

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/flashinfer/decode.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def compile_batch_decode_module(
6262
*args,
6363
verbose: bool = False,
6464
):
65-
uri, path = gen_batch_decode_cu(*args),
65+
uri, path = gen_batch_decode_cu(*args)
6666
return load_cuda_ops(
6767
uri, [path],
6868
verbose=verbose,

0 commit comments

Comments
 (0)