Move value to device before logging for metric

[eunwoosh]

Summary

This is for #3635
and also to deal with Lightning-AI/pytorch-lightning#18803
There is a bug that when updating a value on cpu to metric instance from torchmetric, error is raised during distributed training.
To fix it, I added code to move value to current device explicitly.

How to test

Checklist

• I have added unit tests to cover my changes.​
• I have added integration tests to cover my changes.​
• I have ran e2e tests and there is no issues.
• I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).​
• I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
• I have linked related issues.

License

• I submit my code changes under the same Apache License that covers the project.
  Feel free to contact the maintainers if that's a concern.
• I have updated the license header for each file (see an example below).

# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0