diff --git a/torchrec/distributed/types.py b/torchrec/distributed/types.py
index 95d470af7..40a0e64dd 100644
--- a/torchrec/distributed/types.py
+++ b/torchrec/distributed/types.py
@@ -610,6 +610,7 @@ class KeyValueParams:
     gather_ssd_cache_stats: Optional[bool] = None
     stats_reporter_config: Optional[TBEStatsReporterConfig] = None
     use_passed_in_path: bool = True
+    l2_cache_size: Optional[int] = None
 
     # Parameter Server (PS) Attributes
     ps_hosts: Optional[Tuple[Tuple[str, int], ...]] = None
@@ -623,13 +624,15 @@ def __hash__(self) -> int:
                 self.ssd_storage_directory,
                 self.ssd_rocksdb_write_buffer_size,
                 self.ssd_rocksdb_shards,
-                self.gather_ssd_cache_stats,
-                self.stats_reporter_config,
                 # Parameter Server (PS) Attributes
                 self.ps_hosts,
                 self.ps_client_thread_num,
                 self.ps_max_key_per_request,
                 self.ps_max_local_index_length,
+                # tbe attributes
+                self.gather_ssd_cache_stats,
+                self.stats_reporter_config,
+                self.l2_cache_size,
             )
         )