• Tong WU's avatar
    [Enhancement] Add support for symbolic dimensions in Cython kernel adapter and... · cc00fb65
    Tong WU authored
    [Enhancement] Add support for symbolic dimensions in Cython kernel adapter and improve static shape validation in wrapper (#1024)
    
    * [Enhancement] Add support for symbolic dimensions in Cython kernel adapter and improve static shape validation in wrapper
    
    * [BugFix] Fix shape mismatch and deprecate `T.if()` in fused_moe example
    
    * [Fix] Add `is_symbolic_expr` function to check for symbolic expressions in TIR
    
    - Introduced a new utility function `is_symbolic_expr` to determine if an expression is a symbolic expression, enhancing type checking capabilities.
    - Updated shape handling in `CythonKernelAdapter` to utilize the new function, improving handling for symbolic shapes.
    cc00fb65
example_fusedmoe_tilelang.py 23.3 KB