Fix fairspeq unit test

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/897

Differential Revision: D18250587

Pulled By: myleott

fbshipit-source-id: b9cef376bc014b68766229aab7b6e454480757d3

fairseq/modules/multihead_attention.py

@@ -63,9 +63,12 @@ class MultiheadAttention(nn.Module):
         else:
             self.enable_torch_version = False
 
+    @property
+    def in_proj_weight(self):
+        return torch.cat((self.q_proj.weight, self.k_proj.weight, self.v_proj.weight))
+
     @property
     def in_proj_bias(self):
-        # TODO: Remove this backward compatibility code (in_proj_bias)
         return torch.cat((self.q_proj.bias, self.k_proj.bias, self.v_proj.bias))
 
     def prepare_for_onnx_export_(self):