Fix skipped tests in MultiRankDMPDynamicShardingTest

Summary:
Fix skipped tests. See more context in D78355780.

https://www.internalfb.com/intern/test/562950182314458?ref_report_id=0

Differential Revision: D78583353

Author: jd7-tr

torchrec/distributed/tests/test_dynamic_sharding.py

@@ -583,11 +583,10 @@ def test_sharding(
         """
         Tests resharding from DMP module interface, rather than EBC level.
         """
-        if (
-            self.device == torch.device("cpu")
-            and kernel_type != EmbeddingComputeKernel.FUSED.value
-        ):
-            self.skipTest("CPU does not support uvm.")
+        assume(
+            self.device != torch.device("cpu")
+            or kernel_type == EmbeddingComputeKernel.FUSED.value
+        )
 
         assume(
             sharder_type == SharderType.EMBEDDING_BAG_COLLECTION.value