Added option to set batch size, local workgroup size, and number of iterations for GPU processing
Bug fixes

Don't forget, data is in row major order!