Support memory allocation primitive (ao-bench can not execute)