• Lei Wang's avatar
    [Enhancement] Support float variable as arguments (#250) · 2d0c4169
    Lei Wang authored
    * [Enhancement] Add matrix multiplication functions for integer and float variables in Cython JIT
    
    - Introduced `matmul_int_variable` and `matmul_float_variable` functions to support matrix multiplication with dynamic shapes and additional parameters.
    - Implemented corresponding `run_matmul_int_variable` and `run_matmul_float_variable` functions for testing.
    - Updated test cases to validate the new matrix multiplication implementations.
    - Enhanced error handling in library initialization and compilation processes across various modules.
    - Improved dynamic memory handling in CUDA kernel initialization to provide better error reporting.
    
    * lint fix
    
    * optimize
    2d0c4169
test_tilelang_jit_gemm_cython.py 17.9 KB