commit 58cd133f7ecf664ce869bd0bb480c214bbe60ed8
author Edward Z. Yang <ezyang@mit.edu> Fri Mar 02 12:21:15 2018 -0500
committer GitHub <noreply@github.com> Fri Mar 02 12:21:15 2018 -0500
tree a947c3a50b9ceabb376c78e147735708f54593c0
parent f064c5aa33483061a48994608d890b968ae53fb5

Avoid OOM when running ASAN by splitting nn tests. (#5523)

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

test/run_test.sh

diff --git a/test/run_test.sh b/test/run_test.sh
index 6a42f5f..eec76e7 100755
--- a/test/run_test.sh
+++ b/test/run_test.sh
@@ -33,7 +33,11 @@
 $PYCMD test_sparse.py $@
 
 echo "Running nn tests"
-$PYCMD test_nn.py $@
+# These tests are split up in order to avoid OOMing when
+# we have ASAN, see https://github.com/pytorch/pytorch/issues/5522
+$PYCMD test_nn.py $@ TestNN
+$PYCMD test_nn.py $@ TestNNInit
+$PYCMD test_nn.py $@ PackedSequenceTest
 
 echo "Running legacy nn tests"
 $PYCMD test_legacy_nn.py $@

test/test_nn.py

diff --git a/test/test_nn.py b/test/test_nn.py
index a0442b6..aca31dc 100644
--- a/test/test_nn.py
+++ b/test/test_nn.py
@@ -37,6 +37,11 @@
     from scipy import stats
 
 
+# WARNING: If you add a new top-level test case to this file, you MUST
+# update test/run_test.sh to list it, otherwise it will NOT be run in
+# CI.
+
+
 class PackedSequenceTest(TestCase):
 
     _type_by_name = {