Fix cache modifiers unittests bug

[mawad-amd]

FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-None] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-.wb] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-.cg] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-.cs] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[None-.wt] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-None] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-.wb] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-.cg] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-.cs] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[-.wt] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-None] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-.wb] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-.cg] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-.cs] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.ca-.wt] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
FAILED tests/unittests/test_copy_cache_modifiers.py::test_copy_cache_modifiers[.cg-None] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cg-.cs] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cg-.wt] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
_____________________ test_copy_cache_modifiers[.cv-None] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
_______________________ test_copy_cache_modifiers[.cv-] ________________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cv-.wb] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cv-.cg] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cv-.cs] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!
______________________ test_copy_cache_modifiers[.cv-.wt] ______________________
tests/unittests/test_copy_cache_modifiers.py:102: in test_copy_cache_modifiers
assert torch.allclose(results[i], torch.full((BLOCK_SIZE,), expected_value, dtype=torch.float32)), (
E RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cpu!