Optimised kernel, taking row of A into private memory