sdiehl / numpile Goto Github PK

In [2]: from numpile import autojit

In [3]: %cpaste
Pasting code; enter '--' alone on the line to stop or use Ctrl-D.
:@autojit
:def dot(a, b):
:    c = 0
:    n = a.shape[0]
:    for i in range(n):
:       c += a[i]*b[i]
:    return c
:--

In [4]: %cpaste
Pasting code; enter '--' alone on the line to stop or use Ctrl-D.
:a = np.array(range(1000,2000), dtype='int32')
:b = np.array(range(3000,4000), dtype='int32')
:
:print dot(a,b)
:--
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-4-c2f171281c61> in <module>()
      2 b = np.array(range(3000,4000), dtype='int32')
      3 
----> 4 print dot(a,b)

/Users/ely_spears/scratch/python/numpile/numpile.py in _wrapper(*args)
    956                 return function_cache[key](*args)
    957             else:
--> 958                 llfunc = codegen(ast, specializer, retty, argtys)
    959                 pyfunc = wrap_module(argtys, llfunc)
    960                 function_cache[key] = pyfunc

/Users/ely_spears/scratch/python/numpile/numpile.py in codegen(ast, specializer, retty, argtys)
    976 def codegen(ast, specializer, retty, argtys):
    977     cgen = LLVMEmitter(specializer, retty, argtys)
--> 978     mod = cgen.visit(ast)
    979     cgen.function.verify()
    980 

/Users/ely_spears/scratch/python/numpile/numpile.py in visit(self, node)
    795         name = "visit_%s" % type(node).__name__
    796         if hasattr(self, name):
--> 797             return getattr(self, name)(node)
    798         else:
    799             return self.generic_visit(node)

/Users/ely_spears/scratch/python/numpile/numpile.py in visit_Fun(self, node)
    686         # Setup the register for return type.
    687         if rettype is not void_type:
--> 688             self.locals['retval'] = self.builder.alloca(rettype, "retval")
    689 
    690         map(self.visit, node.body)

/Users/ely_spears/anaconda/envs/basic27/lib/python2.7/site-packages/llvm/core.pyc in alloca(self, ty, size, name)
   2323 
   2324     def alloca(self, ty, size=None, name=""):
-> 2325         sizeptr = size._ptr if size else None
   2326         return _make_value(self._ptr.CreateAlloca(ty._ptr, sizeptr, name))
   2327 

AttributeError: 'str' object has no attribute '_ptr'