_from_host_memory(): use numpy.array()

python/cudf/cudf/core/buffer/buffer.py

@@ -122,13 +122,15 @@ def _from_host_memory(cls: Type[T], data: Any) -> T:
         """Create a Buffer from a buffer or array like object
 
         Data must implement `__array_interface__`, the buffer protocol, and/or
-        be convertible to a buffer object using `numpy.asarray()`
+        be convertible to a buffer object using `numpy.array()`
 
         The host memory is copied to a new device allocation.
 
+        Raises ValueError if array isn't C-contiguous.
+
         Parameters
         ----------
-        data : array-like or buffer-like
+        data : Any
             An object that represens host memory.
 
         Returns
@@ -137,8 +139,10 @@ def _from_host_memory(cls: Type[T], data: Any) -> T:
             Buffer representing a copy of `data`.
         """
 
+        # Convert to numpy array, this will not copy data in most cases.
+        ary = numpy.array(data, copy=False, subok=True)
         # Extract pointer and size
-        ptr, size = get_ptr_and_size(numpy.asarray(data).__array_interface__)
+        ptr, size = get_ptr_and_size(ary.__array_interface__)
         # Copy to device memory
         buf = rmm.DeviceBuffer(ptr=ptr, size=size)
         # Create from device memory

python/cudf/cudf/core/buffer/utils.py

@@ -21,6 +21,10 @@ def as_buffer(
     **not** copied, instead the new buffer keeps a reference to `data` in order
     to retain its lifetime.
 
+    If `data` is an integer, it is assumed to point to device memory.
+
+    Raises ValueError if data isn't C-contiguous.
+
     Parameters
     ----------
     data : int or buffer-like or array-like