The GPU part of HW2 (CompSci)