check use_virtual_table attribute (#3000)

Summary:
Pull Request resolved: #3000

before call table.use_virtual_table check if the table has this attribute first, since some test models might not have the latest code

Reviewed By: kausv

Differential Revision: D75427816

fbshipit-source-id: af6e73f4eac14f98567f56fb0b94115fead1b9c6

Author: emlin

torchrec/quant/embedding_modules.py

@@ -379,7 +379,10 @@ def __init__(
             if table.name in table_names:
                 raise ValueError(f"Duplicate table name {table.name}")
             table_names.add(table.name)
-            key = (table.pooling, table.use_virtual_table)
+            if hasattr(table, "use_virtual_table"):
+                key = (table.pooling, table.use_virtual_table)
+            else:
+                key = (table.pooling, False)
             # pyre-ignore
             self._key_to_tables[key].append(table)
 
@@ -781,7 +784,10 @@ def __init__(  # noqa C901
                     + f" Violating case: {table.name}'s embedding_dim {table.embedding_dim} !="
                     + f" {self._embedding_dim}"
                 )
-            key = (table.data_type, table.use_virtual_table)
+            if hasattr(table, "use_virtual_table"):
+                key = (table.data_type, table.use_virtual_table)
+            else:
+                key = (table.data_type, False)
             self._key_to_tables[key].append(table)
         self._feature_splits: List[int] = []
         for key, emb_configs in self._key_to_tables.items():