commit 79bcf6e5ba96fc9e36782b76a34909bd816a6dca
author James Reed <jamesreed@fb.com> Mon Sep 09 18:58:34 2019 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Mon Sep 09 19:00:35 2019 -0700
tree f90eac431bc5bd82adf9a4b7d725638d1640e965
parent 20204d1fe799cdd1231930ddd70cdc6f7ac5395f

Test scripting and tracing for dynamic linear modules

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25870

Test Plan: Imported from OSS

Differential Revision: D17275747

Pulled By: jamesr66a

fbshipit-source-id: ed8eaf7e9af3127c987e56d17d60b52d039d5ae8

test/test_quantization.py

diff --git a/test/test_quantization.py b/test/test_quantization.py
index 6af15a7..d85d2a0 100644
--- a/test/test_quantization.py
+++ b/test/test_quantization.py
@@ -302,6 +302,7 @@
 
         def checkQuantized(model):
             self.checkDynamicQuantizedLinear(model.fc1)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)
 
@@ -325,6 +326,7 @@
         def checkQuantized(model):
             self.assertEqual(type(model.fc1), torch.nn.Linear)
             self.checkDynamicQuantizedLinear(model.fc2)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)
 
@@ -350,6 +352,7 @@
             self.checkDynamicQuantizedLinear(model.fc3)
             self.checkDynamicQuantizedLinear(model.sub2.fc1)
             self.checkLinear(model.sub2.fc2)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)
 
@@ -376,6 +379,7 @@
             self.checkDynamicQuantizedLinear(model.sub2.fc1)
             self.checkDynamicQuantizedLinear(model.sub2.fc2)
             self.checkDynamicQuantizedLinear(model.fc3)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)
 
@@ -406,6 +410,7 @@
             self.checkDynamicQuantizedLinear(model.sub2.fc1)
             self.checkDynamicQuantizedLinear(model.sub2.fc2)
             self.checkDynamicQuantizedLinear(model.fc3)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)
 
@@ -434,6 +439,7 @@
             self.checkLinear(model.sub2.fc1)
             self.checkDynamicQuantizedLinear(model.sub2.fc2)
             test_only_eval_fn(model, self.calib_data)
+            self.checkScriptable(model, self.calib_data, check_save_load=True)
 
         checkQuantized(model)