Merge pull request #3933 from nouiz/profile

theano/compile/tests/test_profiling.py

@@ -54,13 +54,16 @@ class Test_profiling(unittest.TestCase):
             lines1 = [l for l in the_string.split("\n") if "Max if linker" in l]
             lines2 = [l for l in the_string.split("\n") if "Minimum peak" in l]
             if theano.config.device == 'cpu':
-                assert "Max if linker=cvm(default): 4112KB (8204KB)" in the_string, (
-                    lines1, lines2)
+                assert "CPU: 4112KB (8204KB)" in the_string, (lines1, lines2)
+                assert "CPU: 8204KB (12296KB)" in the_string, (lines1, lines2)
+                assert "CPU: 8208KB" in the_string, (lines1, lines2)
                 assert "Minimum peak from all valid apply node order is 4104KB" in the_string, (
                     lines1, lines2)
             else:
-                assert "Max if linker=cvm(default): 8220KB (8220KB)" in the_string, (
-                    lines1, lines2)
+                assert "CPU: 16KB (16KB)" in the_string, (lines1, lines2)
+                assert "GPU: 8204KB (8204KB)" in the_string, (lines1, lines2)
+                assert "GPU: 12300KB (12300KB)" in the_string, (lines1, lines2)
+                assert "GPU: 8212KB" in the_string, (lines1, lines2)
                 assert "Minimum peak from all valid apply node order is 4116KB" in the_string, (
                     lines1, lines2)