made grad method preserve partialSum
made unit tests try different partial sums
fixed size of nv_partialsum
removed warning that we don't use partialsum
moved an assert to the last possible branch
test passes