Commit ca158ae9 authored by flyingdown's avatar flyingdown
Browse files

fix get config assert error

parent 8bc2cc01
......@@ -1993,7 +1993,9 @@ def fused_experts_impl(
)
# debug
# print(f"M:{M}, E:{E}, N1:{N1}, N2:{N2}, K2:{K2}, top_k_num:{top_k_num}, block_shape:{block_shape}, dtype:{hidden_states.dtype}, status:{status}")
assert status, "config not found."
# assert status, "config not found."
if not status:
config = get_config_func(M)
# We can reuse the memory between these because by the time we need
# cache3, we're done with cache1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment