Temporarily disable rmm devicebuffer in array.py (#4333)

Temporary fix for #4332

Authors:
  - Dante Gama Dessavre (https://github.com/dantegd)

Approvers:
  - William Hicks (https://github.com/wphicks)
  - John Zedlewski (https://github.com/JohnZed)

URL: #4333

python/cuml/common/array.py

@@ -19,7 +19,9 @@
 import operator
 import nvtx
 
-from rmm import DeviceBuffer
+# Temporarily disabled due to CUDA 11.0 issue
+# https://github.com/rapidsai/cuml/issues/4332
+# from rmm import DeviceBuffer
 from cudf import DataFrame
 from cudf import Series
 from cudf.core.buffer import Buffer
@@ -383,7 +385,10 @@ def _check_low_level_type(data):
     elif not (
         hasattr(data, "__array_interface__")
         or hasattr(data, "__cuda_array_interface__")
-    ) or isinstance(data, (DeviceBuffer, Buffer)):
+        # Temporarily disabled due to CUDA 11.0 issue
+        # https://github.com/rapidsai/cuml/issues/4332
+        # ) or isinstance(data, (DeviceBuffer, Buffer)):
+    ) or isinstance(data, Buffer):
         return True
     else:
         return False

python/cuml/test/test_array.py

@@ -30,7 +30,9 @@
 from cuml.common.array import CumlArray
 from cuml.common.memory_utils import _get_size_from_shape
 from cuml.common.memory_utils import _strides_to_order
-from rmm import DeviceBuffer
+# Temporarily disabled due to CUDA 11.0 issue
+# https://github.com/rapidsai/cuml/issues/4332
+# from rmm import DeviceBuffer
 
 if sys.version_info < (3, 8):
     try:
@@ -283,7 +285,9 @@ def test_create_empty(shape, dtype, order):
     else:
         assert ary.shape == shape
     assert ary.dtype == np.dtype(dtype)
-    assert isinstance(ary._owner.data.mem._owner, DeviceBuffer)
+    # Temporarily disabled due to CUDA 11.0 issue
+    # https://github.com/rapidsai/cuml/issues/4332
+    # assert isinstance(ary._owner.data.mem._owner, DeviceBuffer)
 
 
 @pytest.mark.parametrize('shape', test_shapes)