HW4 problem1 pycuda