add tests
add gpu tests