test scriptify MultiheadAttention

Summary: We introduce a new test suite where we will iteratively ensure that various model components are TorchScript compliant (JIT-able).

Differential Revision: D18799715

fbshipit-source-id: 9e1fa1cdb3d0bdc5aa6434eb4b57c9724b35e2a0

Author: jhcross

pytorch_translate/test/test_export_models.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+
+import unittest
+
+import torch
+from fairseq.modules import multihead_attention
+
+
+class TestExportModels(unittest.TestCase):
+    @unittest.skip("TDD: placeholder for development")
+    def test_export_multihead_attention(self):
+        module = multihead_attention.MultiheadAttention(embed_dim=8, num_heads=2)
+        torch.jit.script(module)