zram: rewrite zram to make it compatible with zsmalloc