rapidsai / rapidsai/deployment

Numba 0.61.0 broken on Google Colab (Feb 25)

Open
#528 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

bug platform/colab
Dominant language
Jupyter Notebook
Stars
15
Forks
41
Avg merge
1d 8h
Merged PRs (30d)
6

Description

As of Feb 13th 2025 the latest numba release 0.61.0 is not compatible with the Google Colab GPU runtime.

Reproducer

The first numba cell fails

import numpy as np
from numba import cuda

data = np.asarray(range(10))
output = np.zeros(len(data))

@cuda.jit
def foo(input_array, output_array):
    i = cuda.grid(1)
    output_array[i] = input_array[i]
    
foo[1, len(data)](data, output)

output
/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/dispatcher.py:536: NumbaPerformanceWarning: Grid size 1 will likely result in GPU under-utilization due to low occupancy.
  warn(NumbaPerformanceWarning(msg))
ERROR:numba.cuda.cudadrv.driver:Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION

---------------------------------------------------------------------------

CudaAPIError                              Traceback (most recent call last)

[/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/cudadrv/driver.py](https://localhost:8080/#) in add_ptx(self, ptx, name)
   2806         try:
-> 2807             driver.cuLinkAddData(self.handle, enums.CU_JIT_INPUT_PTX,
   2808                                  ptxbuf, len(ptx), namebuf, 0, None, None)

10 frames

CudaAPIError: [222] Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION


During handling of the above exception, another exception occurred:

LinkerError                               Traceback (most recent call last)

[/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/cudadrv/driver.py](https://localhost:8080/#) in add_ptx(self, ptx, name)
   2808                                  ptxbuf, len(ptx), namebuf, 0, None, None)
   2809         except CudaAPIError as e:
-> 2810             raise LinkerError("%s\n%s" % (e, self.error_log))
   2811 
   2812     def add_file(self, path, kind):

LinkerError: [222] Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION
ptxas application ptx input, line 9; fatal   : Unsupported .version 8.5; current version is '8.4'

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the linked NVIDIA accelerated-computing-hub notebook and reproduce the first Numba CUDA cell on a Google Colab GPU runtime. Compare the installed Numba and PTX versions with the runtime's supported version, then identify the deployment or notebook change needed. Done means the example runs successfully without CUDA_ERROR_UNSUPPORTED_PTX_VERSION.

Written by the indexing model from the issue text.

Assessment

Tech stack
jupyter-notebook, python
Domain
cloud, documentation
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
42/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.