• Thomas Ning's avatar
    Ck tile gemm example (#1488) · caacd388
    Thomas Ning authored
    
    
    * Checkpoint: Finished with the tile example & kernel verification, working on the different matrix layout
    
    * Finished the Matrix Layout feature set up. Note: Need to modify the inner block to solve the shuffle problem in the future.
    
    * Fix: Clang Format, API fixed from fmha
    
    * fix with better naming convention
    
    * revert back the pipeline code of fmha
    
    * Fixed: Addressed the comments and merge the GEMM shape of GEMM Operator and FMHA Operator to one.
    
    * clang format with the reference_gemm file
    
    * convert the clang format with the remod.py
    
    * Changed the format and variable name of the kernel gemm_shape and partitioner
    
    ---------
    Co-authored-by: default avatarthomasning <thomasning@banff-cyxtera-s70-4.ctr.dcgpu>
    caacd388
gemm.hpp 2.37 KB