OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 |
| 3 # Copyright (c) 2013 Google Inc. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. |
| 6 |
| 7 """ |
| 8 Make sure PGO is working properly. |
| 9 """ |
| 10 |
| 11 import TestGyp |
| 12 |
| 13 import os |
| 14 import sys |
| 15 |
| 16 if sys.platform == 'win32': |
| 17 test = TestGyp.TestGyp(formats=['msvs', 'ninja']) |
| 18 |
| 19 CHDIR = 'linker-flags' |
| 20 test.run_gyp('pgo.gyp', chdir=CHDIR) |
| 21 |
| 22 def IsPGOAvailable(): |
| 23 """Returns true if the Visual Studio available here supports PGO.""" |
| 24 test.build('pgo.gyp', 'gen_linker_option', chdir=CHDIR) |
| 25 tmpfile = test.read(test.built_file_path('linker_options.txt', chdir=CHDIR)) |
| 26 return any(line.find('PGOPTIMIZE') for line in tmpfile) |
| 27 |
| 28 # Test generated build files look fine. |
| 29 if test.format == 'ninja': |
| 30 ninja = test.built_file_path('obj/test_pgo_instrument.ninja', chdir=CHDIR) |
| 31 test.must_contain(ninja, '/LTCG:PGINSTRUMENT') |
| 32 test.must_contain(ninja, 'test_pgo.pgd') |
| 33 ninja = test.built_file_path('obj/test_pgo_optimize.ninja', chdir=CHDIR) |
| 34 test.must_contain(ninja, '/LTCG:PGOPTIMIZE') |
| 35 test.must_contain(ninja, 'test_pgo.pgd') |
| 36 ninja = test.built_file_path('obj/test_pgo_update.ninja', chdir=CHDIR) |
| 37 test.must_contain(ninja, '/LTCG:PGUPDATE') |
| 38 test.must_contain(ninja, 'test_pgo.pgd') |
| 39 elif test.format == 'msvs': |
| 40 LTCG_FORMAT = '<LinkTimeCodeGeneration>%s</LinkTimeCodeGeneration>' |
| 41 vcproj = test.workpath('linker-flags/test_pgo_instrument.vcxproj') |
| 42 test.must_contain(vcproj, LTCG_FORMAT % 'PGInstrument') |
| 43 test.must_contain(vcproj, 'test_pgo.pgd') |
| 44 vcproj = test.workpath('linker-flags/test_pgo_optimize.vcxproj') |
| 45 test.must_contain(vcproj, LTCG_FORMAT % 'PGOptimization') |
| 46 test.must_contain(vcproj, 'test_pgo.pgd') |
| 47 vcproj = test.workpath('linker-flags/test_pgo_update.vcxproj') |
| 48 test.must_contain(vcproj, LTCG_FORMAT % 'PGUpdate') |
| 49 test.must_contain(vcproj, 'test_pgo.pgd') |
| 50 |
| 51 # When PGO is available, try building binaries with PGO. |
| 52 if IsPGOAvailable(): |
| 53 pgd_path = test.built_file_path('test_pgo.pgd', chdir=CHDIR) |
| 54 |
| 55 # Test if 'PGInstrument' generates PGD (Profile-Guided Database) file. |
| 56 if os.path.exists(pgd_path): |
| 57 test.unlink(pgd_path) |
| 58 test.must_not_exist(pgd_path) |
| 59 test.build('pgo.gyp', 'test_pgo_instrument', chdir=CHDIR) |
| 60 test.must_exist(pgd_path) |
| 61 |
| 62 # Test if 'PGOptimize' works well |
| 63 test.build('pgo.gyp', 'test_pgo_optimize', chdir=CHDIR) |
| 64 test.must_contain_any_line(test.stdout(), ['profiled functions']) |
| 65 |
| 66 # Test if 'PGUpdate' works well |
| 67 test.build('pgo.gyp', 'test_pgo_update', chdir=CHDIR) |
| 68 # With 'PGUpdate', linker should not complain that sources are changed after |
| 69 # the previous training run. |
| 70 test.touch(test.workpath('linker-flags/inline_test_main.cc')) |
| 71 test.unlink(test.built_file_path('test_pgo_update.exe', chdir=CHDIR)) |
| 72 test.build('pgo.gyp', 'test_pgo_update', chdir=CHDIR) |
| 73 test.must_contain_any_line(test.stdout(), ['profiled functions']) |
| 74 |
| 75 test.pass_test() |
OLD | NEW |