Fixes for 6.3-rc2:

 * Add a few debugging assertions so that people (me) trying to port
   code to the new allocator functions don't mess up the caller
   requirements.

Signed-off-by: Darrick J. Wong <djwong@kernel.org>