Merge pull request #257 from jakirkham/drop_numba_sizeof_code

Drop Numba `DeviceNDArray` code for `sizeof`
rapidsai · Mar 19, 2020 · bf6f868 · bf6f868
2 parents ba6829e + b9bc27e
commit bf6f868
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 14 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -5,6 +5,7 @@
 - Respect `temporary-directory` config for spilling (#247) `John Kirkham`_
 - Relax CuPy pin (#248) `John Kirkham`_
 - Added `ignore_index` argument to `partition_by_hash()` (#253) `Mads R. B. Kristensen`_
+- Drop Numba `DeviceNDArray` code for `sizeof` (#257) `John Kirkham`_
 
 0.12
 ----

diff --git a/dask_cuda/device_host_file.py b/dask_cuda/device_host_file.py
@@ -1,7 +1,6 @@
 import os
 
 import dask
-from dask.sizeof import sizeof
 from distributed.protocol import (
     dask_deserialize,
     dask_serialize,
@@ -25,18 +24,6 @@
     import numba.cuda as cuda_memory_manager
 
 
-# Register sizeof for Numba DeviceNDArray while Dask doesn't add it
-if not hasattr(sizeof, "register_numba"):
-
-    @sizeof.register_lazy("numba")
-    def register_numba():
-        import numba
-
-        @sizeof.register(numba.cuda.cudadrv.devicearray.DeviceNDArray)
-        def sizeof_numba_devicearray(x):
-            return int(x.nbytes)
-
-
 class DeviceSerialized:
     """ Store device object on the host
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-dask>=2.4.0
+dask>=2.9.0
 distributed>=2.7.0
 pynvml>=8.0.3
 numpy>=1.16.0