These checks are causing issues for very large USM allocations because
the `MAX_MEM_ALLOC_SIZE` reported is lower than what CUDA actually
supports.

We will follow up with an update on the reported `MAX_MEM_ALLOC_SIZE`,
but it makes sense to remove the checks either way, as the CUDA
allocation functions will return an error if they can't allocate the
memory.