• Lei Wang's avatar
    [Refactor] Generalize fp8 process (#1372) · 92121fc6
    Lei Wang authored
    * [Refactor] Update condition for benchmarking in example_gemv.py and simplify cached library path handling in sparse.py
    
    * [Enhancement] Extend support for float8 data types in GEMM operations
    
    - Updated GEMM operations to recognize additional float8 data types: `float8_e4m3fn` and `float8_e5m2fnuz`.
    - Refactored condition checks in `checkWgmma` methods to simplify float8 type handling.
    - Adjusted test cases to ensure compatibility with the new float8 types in tile language examples.
    
    * lint fix
    92121fc6
copy.cc 83.2 KB