pyopencl working for square matrices