Moving data classes to types file.

Summary: Part of refactor to make lp_planner.py smaller, part 1, moving rank and shard info to types.py

Differential Revision: D72186211

fbshipit-source-id: 501e20689446ed75335cd48b825713611a66a56f

Author: Tony Zhang

torchrec/distributed/planner/types.py

@@ -342,6 +342,27 @@ def __str__(self) -> str:
         return f"Shard size: {tuple(self.size)}, offset: {tuple(self.offset)}, storage: {str(self.storage)}, perf: {str(self.perf)}, rank: {self.rank}"
 
 
+@dataclass
+class Rank:
+    key: str
+    hbm: float
+    ddr: float
+    node_key: str
+
+
+@dataclass
+class ShardInfo:
+    key: str
+    sharding_type: str
+    compute_kernel: str
+    table_name: str
+    rank_key: str  # either a number or DUMMY_RANK_KEY
+    shard: Shard
+    module_fqn: str
+    _index_in_shards: int
+    _index_in_search_space: int
+
+
 class ShardingOption:
     """
     One way of sharding an embedding table. In the enumerator, we generate