new test_profiling.py

theano/compile/profiling.py

@@ -737,16 +737,16 @@ class ProfileStats(object):
                 for i in range(len(node_list)):
                     v = node_list[i:i+1]
                     if check_node_state(v[0]):
-                        for i in v[0].outputs:
-                            compute_map[i][0] = 1 
+                        for node in v[0].outputs:
+                            compute_map[node][0] = 1 
                         if len(node_list) == 1:
                             yield v
                         else:
                             rest = node_list[ :i] + node_list[i+1: ]
                             for p in min_memory_generator(rest, compute_map):
                                 yield v+p
-                        for i in v[0].outputs:
-                            compute_map[i][0] = 0
+                        for node in v[0].outputs:
+                            compute_map[node][0] = 0
 
             min_order = []
 

theano/compile/tests/test_profiling.py

@@ -15,20 +15,34 @@ def test_profiling():
         theano.config.profile = True
         theano.config.profile_memory = True
 
-        x = T.dvector("x")
-        y = T.dvector("y")
-        z = x + y
+        val1 = T.dvector("val1")
+        val2 = T.dvector("val2")
+        val3 = T.dvector("val3")
+        val4 = T.dvector("val4")
+        val5 = T.dvector("val5")
+        val6 = T.dvector("val6")
+        val7 = T.dvector("val7")
+        val8 = T.dvector("val8")
+        val9 = T.dvector("val9")
+        x = [val1, val2, val3, val4, val5, val6, val7, val8, val9]
+
+        z = [x[i] + x[i+1] for i in range(len(x)-1)] + [T.outer(x[i], x[i+1]).sum() for i in range(len(x)-1)]
+
         p = theano.ProfileStats(False)
+
         if theano.config.mode in ["DebugMode", "DEBUG_MODE"]:
             m = "FAST_RUN"
         else:
             m = None
-        f = theano.function([x, y], z, profile=p, name="test_profiling",
+
+        f = theano.function([val1, val2, val3, val4, val5, val6, val7, val8, val9], z, profile=p, name="test_profiling",
                             mode=m)
-        output = f([1, 2, 3, 4], [1, 1, 1, 1])
+
+        output = f([0, 1, 2, 3, 4], [1, 2, 3, 4, 5], [2, 3, 4, 5, 6], [3, 4, 5, 6, 7], [4, 5, 6, 7, 8], [5, 6, 7, 8, 9], [6, 7, 8, 9, 10], [7, 8, 9, 10, 11], [8, 9, 10, 11, 12])
 
         buf = StringIO.StringIO()
         f.profile.summary(buf)
+
     finally:
         theano.config.profile = old1
         theano.config.profile_memory = old2