Revert changes to device_malloc calls; overload instead.