Factors: 0 (moe_topk_softmax), 2 (moe_fused_gemm), 5 (gdn_chunk_fwd) Fixes: topk_softmax fallback (2304x/token), GDN NaN (frac=0.9998-1.0)
4 lines
82 B
Python
4 lines
82 B
Python
from .ex_loader import EXEngine, get_engine
|
|
|
|
__all__ = ["EXEngine", "get_engine"]
|