nvme: Refactor geom setting to function.
Refactor setting of geometry for the disk to its own function. No
functional changes.
Reviewed by: imp
Differential Revision: https://reviews.freebsd.org/D33032
(cherry picked from commit dffd882d12d2a71aca464f48209ec9ae6f393b15)