Implement rewriting of assignments into loopy kernels