• Lei Wang's avatar
    [Enhancement] Add new matrix multiplication functions and tests for GEMM with... · 9e5a757e
    Lei Wang authored
    [Enhancement] Add new matrix multiplication functions and tests for GEMM with transpose options (#331)
    
    - Introduced `matmul_rs` function for flexible matrix multiplication with optional transposition.
    - Added `run_gemm_rs` function to facilitate testing of the new matrix multiplication implementation.
    - Expanded test coverage for GEMM with additional cases for transposition configurations.
    - Corrected index usage in `gemm.h` to ensure proper matrix layout handling.
    
    These changes enhance the GEMM functionality and improve testing capabilities for various matrix configurations.
    9e5a757e
test_tilelang_test_amd.py 5.99 KB