update the test assert when device=gpu

theano/compile/tests/test_profiling.py

@@ -60,8 +60,10 @@ class Test_profiling(unittest.TestCase):
                 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)