Minor cange to matMulDiag kernel.
Fixed invalid grid sizes in some calculations.
Fixed few matrix/vector sizes.