Use DeviceBuffer in GPU update and NBNXM code