Compatible with rmm api 2.0.0