Skip to content
Go to file

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time

Numba ROC Examples

This repository contains several example of ROC GPU programming using Numba. It is a fork and update of examples written for HSA

WARNING: The examples here and the underlying compilation chain are both under development!

For the purposes of this README, the $ symbol indicates the command prompt.

Set up machine

  • First, install the ROC stack from AMD as per the installation instructions provided here .

  • Create and activate a conda environment (named e.g. amd_roc) as follows:

$ conda create -n amd_roc -c numba numba roctools jupyter bokeh statsmodels \
python=3 h5py -y
$ source activate amd_roc
  • Note that users should be a member of the video group to have access to the GPU.

Run examples

There are two examples working at present, both revolving around kernel density estimation. The first is a Jupyter notebook, multi_variate_kde_example.ipynb which can be launched with the following:

$ jupyter notebook numba_roc_examples/kerneldensity/

You can see the rendered notebook here. The AMD GPU implementation used in notebook can be found here.

The second is a bokeh application that can be launched following the instructions in the of the numba_roc_examples/kde_bokeh directory.

Run Tests

Run the full test suite, it is expected that this will fail (still under development):

$ ./

Selectively run tests from subdirectories:

$ ./ <subdirectory>


$ ./ numba_roc_examples/kerneldensity


Examples for ROC GPUs




No releases published


No packages published
You can’t perform that action at this time.