fix GPU unit test