[Numpy-discussion] Numpy and PEP 343
Tim Hochberg
tim.hochberg at cox.net
Fri Mar 3 10:20:01 CST 2006
David M. Cooke wrote:
>Tim Hochberg <tim.hochberg at cox.net> writes:
>
>
>
>>David M. Cooke wrote:
>>
>>
>
>It's in. Check the setup.py -- I left some gcc'isms in there
>(-funroll-loops). That should be conditional on the compiler.
>
>
I does whine about that, but it ignores it, so I did to ;). The
misplaced declaration of dims I mentioned earlier was the real culprit.
>>This is probably thinking two far ahead, but an interesting thing to
>>try would be adding conditional expressions:
>>
>>c = evaluate("(2*a + b) if (a > b) else (2*b + a)")
>>
>>If that could be made to work, and work fast, it would save both
>>memory and time in those cases where you have to vary the computation
>>based on the value. At present I end up computing the full arrays for
>>each case and then choosing which result to use based on a mask, so it
>>takes three times as much space as doing it element by element.
>>
>>
>
>It's on my list of things to think about, along with implementing
>reductions (sum, in particular). It'd probably look more like
>
>c = evaluate("where(a > b, 2*a+b, 2*b+a)")
>
>because of the vector nature of the code. Doing the "if" elementwise
>would mean the bytecode program would have to be run for each element,
>instead of on a vector of some fixed length.
>
>
That makes sense. One thought I had with respect to the various numpy
functions (sin, cos, pow, etc) was to just have the bytecodes:
call_unary_function, function_id, store_in, source
call_binary_function, function_id, store_in, source1, source2
call_trinary_function, function_id, store_in, source1, source2, source3
Then just store pointers to the functions in relevant tables. In it's
most straightforward form, you'd need 6 character chunks of bytecode
instead of four. However, if that turns out to slow everything else
down I think it could be packed down to 4 again. The function_ids could
probably be packed into the opcode (as long as we stay below 200 or so
functions, which is probably safe), the other way to pack things down is
to require that one of the sources for trinary functions is always a
certain register (say register-0). That requires a bit more cleverness
at the compiler level, but is probably feasible.
OT, stupid subversion question: where do I find the sandbox or
whereever this got checked in?
-tim
More information about the Numpy-discussion
mailing list