• Lei Wang's avatar
    [Language] Refactor reduce and support shared memory as its in/out (#1219) · 47039f06
    Lei Wang authored
    * [Refactor] Update ReduceOpNode to use absolute values in Max computation and remove unused shared memory reduction logic
    
    * Changed Max computation for AbsMax type to use absolute values of lhs and rhs.
    * Removed unused shared memory reduction logic and related checks for buffer dimensions and thread extents, simplifying the Lower method.
    * Added a fatal log for unsupported buffer scope reductions.
    
    * reduce fix
    
    * [Fix] Update type check for eval value in Builder class
    
    * Changed the type check for eval values to raise a TypeError for unsupported types, specifically excluding instances of tvm.tir.Buffer. This improves error handling and clarity in the Builder class.
    47039f06
builder.py 25.6 KB