working properly, but GPU done on CPU