alloc2malloc.cpp

Go to the documentation of this file.

#include "mim/plug/mem/pass/alloc2malloc.h"
 
#include "mim/plug/mem/mem.h"
 
namespace mim::plug::mem {
 
const Def* Alloc2Malloc::rewrite(const Def* def) {
    if (auto alloc = Axm::isa<mem::alloc>(def)) {
        auto [pointee, addr_space] = alloc->decurry()->args<2>();
        return op_malloc(pointee, alloc->arg());
    } else if (auto slot = Axm::isa<mem::slot>(def)) {
        auto [Ta, mi]              = slot->uncurry_args<2>();
        auto [pointee, addr_space] = Ta->projs<2>();
        auto [mem, id]             = mi->projs<2>();
        return op_mslot(pointee, mem, id);
    }
 
    return def;
}
 
} // namespace mim::plug::mem