Improve memory tracking in vulkan memory mapping tests