Anaconda Acclerate / NumbaPro CUDA Linking Error OSX

Question

Overall goal is to use NumbaPro to run some functions on the GPU (on OSX 10.8.3).

Before starting, I just wanted to get everything set up. According to this page I installed CUDA, registered as a CUDA developer, downloaded the Compiler SDK and set up the NUMBAPRO_NVVM=/path/to/libnvvm.dylib environment variable.

However, running this basic test function:

from numbapro import autojit

@autojit(target='gpu')
def my_function(x):
    if x == 0.0:
        return 1.0
    else:
        return x*x*x

print my_function(4.4)
exit()

Brings up this error:

File ".../anaconda/lib/python2.7/site-packages/numba/decorators.py", line 207, in compile_function
    compiled_function = dec(f)
File "...lib/python2.7/site-packages/numbapro/cudapipeline/decorators.py", line 35, in _jit_decorator
File "...lib/python2.7/site-packages/numbapro/cudapipeline/decorators.py", line 128, in __init__
File "...lib/python2.7/site-packages/numbapro/cudapipeline/environment.py", line 31, in generate_ptx
File "...lib/python2.7/site-packages/numbapro/cudapipeline/environment.py", line 186, in _link_llvm_math_intrinsics
KeyError: 1

I've tried @vectorize'ing instead of autojit, same error. @autojit by itself with no target works fine.

Any ideas?

Answer 1

For posterity's sake, I asked Continuum Support. They responded:

It seems that you are running a CUDA GPU with compute capability 1.x. NVVM only supports CC2.0 and above. We definitely should have a better error reporting and make it clear in the NumbaPro documentation for the supported compute capability.