in group_table check string value of pooling mode (pytorch#1124)

Summary:
Pull Request resolved: pytorch#1124

If the model is from torch package, checking pooling mode directly wouldn't result in a match because one would from <torch_package_x>.

Instead, we can check the string value.

Reviewed By: houseroad

Differential Revision: D45059135

fbshipit-source-id: 4b452bbbecc1c6b561ebb21906b017b3540205cd

torchrec/distributed/embedding_sharding.py

@@ -163,7 +163,7 @@ def _group_tables_per_rank(
                                     compute_kernel_type = EmbeddingComputeKernel.QUANT
                                 if (
                                     table.data_type == data_type
-                                    and table.pooling == pooling
+                                    and table.pooling.value == pooling.value
                                     and table.has_feature_processor
                                     == has_feature_processor
                                     and compute_kernel_type == compute_kernel