Initial unit tests for gpu