Add some GPU codegen tests