More modifications to LU kernels: faster for multi-partition case