• Lei Wang's avatar
    [Language] Support tilelang `alloc_var(dtype, init=x)` (#1092) · bddb125e
    Lei Wang authored
    * - carry existing local-var initializer map into OpaqueBlockLower, reattach it to
        generated Allocates and the PrimFunc attrs
      - thread the map through FlattenBuffer and StorageRewrite so flattened/merged
        allocations keep their tl.local_var_init annotations
      - teach annotation handling to accept scalar initializers, resolve buffers, and merge
        with existing stat
    
    * lint fix
    
    * enhance
    
    * lint fix
    
    * lint fix
    bddb125e
codegen_cuda.cc 98.8 KB