Added new gpu code for associative memory.

Should work better with new version of nengo.