diag: Properly update memory device clients

Currently diag driver is not updating memory
device clients properly after logging switch.
This patch fixes this issue.

Change-Id: I037188047c6beceef108a7d5d65c2697f835cd3e
Signed-off-by: Mohit Aggarwal <maggarwa@codeaurora.org>