diff --git a/python/cudf/cudf/__init__.py b/python/cudf/cudf/__init__.py index 2e4c1ccebd5..df09a72ce25 100644 --- a/python/cudf/cudf/__init__.py +++ b/python/cudf/cudf/__init__.py @@ -4,7 +4,7 @@ validate_setup() import cupy -from numba import cuda +from numba import config as numba_config, cuda import rmm @@ -104,5 +104,13 @@ cuda.set_memory_manager(rmm.RMMNumbaManager) cupy.cuda.set_allocator(rmm.rmm_cupy_allocator) +try: + # Numba 0.54: Disable low occupancy warnings + numba_config.CUDA_LOW_OCCUPANCY_WARNINGS = 0 +except AttributeError: + # Numba < 0.54: No occupancy warnings + pass +del numba_config + __version__ = get_versions()["version"] del get_versions