From 15b569d8efaa9cbfc7820e0e6c5c8183b66c6cf9 Mon Sep 17 00:00:00 2001 From: Dante Gama Dessavre Date: Mon, 8 Nov 2021 16:26:22 -0600 Subject: [PATCH] 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: https://github.com/rapidsai/cuml/pull/4333 --- python/cuml/common/array.py | 9 +++++++-- python/cuml/test/test_array.py | 8 ++++++-- 2 files changed, 13 insertions(+), 4 deletions(-) diff --git a/python/cuml/common/array.py b/python/cuml/common/array.py index 33d3657d83..6300f61aa5 100644 --- a/python/cuml/common/array.py +++ b/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 diff --git a/python/cuml/test/test_array.py b/python/cuml/test/test_array.py index ed1191234f..8ceeb069d6 100644 --- a/python/cuml/test/test_array.py +++ b/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)