support quantize fp32 embedding tables (#2794)

Summary: Pull Request resolved: #2794

Reviewed By: aporialiao

Differential Revision: D71762539

Pulled By: iamzainhuda

fbshipit-source-id: d468abc22b96bc40befaf0a8975bf5c06a0b6c41

Author: tiankongdeguiji

torchrec/quant/embedding_modules.py

@@ -252,6 +252,10 @@ def quantize_state_dict(
                     if tensor.dtype == torch.float:
                         tensor = tensor.half()
                     quant_res = tensor.view(torch.uint8)
+                elif data_type == DataType.FP32:
+                    if tensor.dtype == torch.float16:
+                        tensor = tensor.float()
+                    quant_res = tensor.view(torch.uint8)
                 else:
                     quant_res = (
                         torch.ops.fbgemm.FloatOrHalfToFusedNBitRowwiseQuantizedSBHalf(