Skip to content

Commit

Permalink
bugfix: Fix the "not enough values to unpack" error. (#540)
Browse files Browse the repository at this point in the history
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)
```
  • Loading branch information
yuxianq authored Oct 19, 2024
1 parent d0b0b7f commit b878508
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/flashinfer/decode.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ def compile_batch_decode_module(
*args,
verbose: bool = False,
):
uri, path = gen_batch_decode_cu(*args),
uri, path = gen_batch_decode_cu(*args)
return load_cuda_ops(
uri, [path],
verbose=verbose,
Expand Down

0 comments on commit b878508

Please sign in to comment.