Does Numba support for multiple dictionary multiplication in terms of `multi_dot`?

100 views Asked by At

I need to multiply three 2-D matrices of size 25x25, 25x60, and 60x60 to get a result of size 25x60 using numpy. For fast multiplication, I wanted to use from numpy.linalg import multi_dot and also tried to parallely execute it in GPU using @jit(nopython=True) instruction.

However, in my program I receive the following numba error.

TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Untyped global name 'multi_dot': cannot determine Numba type of <class 'function'>

This error is not occuring when I am not using the @jit(nopython=True) command above the function definition i.e. sequentially it is working fine for me.

0

There are 0 answers