add more instantiation for EmbeddingSpMDMAvx2 (pytorch#274)

Summary:
Pull Request resolved: pytorch#274

Solved unresolved references in test/EmbeddingSpMDMTest.cc. Add instantiation for input type uint8_t.

Reviewed By: jspark1105

Differential Revision: D19654528

fbshipit-source-id: 5dca40e1f6530c5720f290c27730e2e13bd9dce7

src/EmbeddingSpMDMAvx2.cc

@@ -173,5 +173,29 @@ template bool EmbeddingSpMDMBlockSize1_(
     float* out,
     bool is_weight_positional);
 
+template bool EmbeddingSpMDMBlockSize1_(
+    const std::int64_t output_size,
+    const std::int64_t index_size,
+    const std::int64_t data_size, // the number of rows in input
+    const std::uint8_t* input,
+    const std::int64_t* indices,
+    const int* lengths,
+    const float* weights, // optional, can be null for non-weighted sum
+    bool normalize_by_lengths,
+    float* out,
+    bool is_weight_positional);
+
+template bool EmbeddingSpMDMBlockSize1_(
+    const std::int64_t output_size,
+    const std::int64_t index_size,
+    const std::int64_t data_size, // the number of rows in input
+    const std::uint8_t* input,
+    const std::int32_t* indices,
+    const int* lengths,
+    const float* weights, // optional, can be null for non-weighted sum
+    bool normalize_by_lengths,
+    float* out,
+    bool is_weight_positional);
+
 } // namespace internal
 } // namespace fbgemm