support metadata mask for feature score eviction

Differential Revision: D80131015

Author: EddyLXJ

torchrec/modules/embedding_configs.py

@@ -212,6 +212,13 @@ class FeatureScoreBasedEvictionPolicy(VirtualTableEvictionPolicy):
     decay_rate: float = 0.99  # default decay by default #TODO: Change to real value
     max_training_id_num_per_rank: int = 0  # max number of training ids per rank
     target_eviction_percent: float = 0.0  # target eviction percent
+    inference_eviction_feature_score_threshold: Optional[float] = (
+        None  # 0 means no eviction
+    )
+
+    def __post_init__(self) -> None:
+        if self.inference_eviction_feature_score_threshold is None:
+            self.inference_eviction_feature_score_threshold = 0
 
 
 @dataclass