small fix
small test change to allow testing on the GPU.
Fix test value when testing with the GPU.
Merge pull request #7 from nouiz/RoyXue-fix_crash_and_GPU_support

Roy xue fix crash and gpu support