[fix] Add back the allocator implementation for WMSDK