Use local memory in X-Sweep update step
Fix/add local memory for update kernels, add unit tests for local memory