Fix up GPU Implementation

Signed-off-by: Chester Kener <Cl3Kener@gmail.com>