• Lei Wang's avatar
    [Refactor] Phaseout vmap for Tile Operators (#1334) · f5d9da46
    Lei Wang authored
    
    
    * Refactor GEMM and Reduce operations by moving NormalizeToBufferRegion and MakeAccessPtrFromRegion to utils.{h,cc} for better code organization and reuse.
    
    * lint fix
    
    * Refactor region handling by removing the RegionOp and updating NormalizeToBufferRegion to only accept BufferLoad and BufferRegion. This change improves code organization and simplifies the handling of memory regions across various operations.
    
    * fix
    
    * Refactor memory region handling by introducing `tl.region` calls across various operations, including GEMM and fill functions. This change enhances the consistency of region management and improves code organization by utilizing utility functions for buffer region conversions.
    
    * fix
    
    * fix
    
    * test fix
    
    * lint fix
    
    * Refactor GEMM operations to improve memory region handling by replacing `mbarPtr_` with `mbarRegion_` and updating related logic in both C++ and Python implementations. This change enhances the clarity and consistency of buffer region management.
    
    * fix
    
    * lint fix
    
    * fix
    
    * fix
    
    * test fix
    
    * lint fix
    
    * lint fix
    
    * minor fix
    
    * fix
    
    ---------
    Co-authored-by: default avatarZhiwen Mo <zm125@ic.ac.uk>
    f5d9da46
reduce.h 5.97 KB