| """Test module that runs llvm-profdata merge after executing the benchmark.""" |
| from litsupport import shellcommand |
| from litsupport import testplan |
| |
| |
| def _mutateCommandline(context, commandline): |
| """Adjust runscript to set a different value to the LLVM_PROFILE_FILE |
| environment variable for each execution.""" |
| profilefile = context.tmpBase + ".profraw" |
| prefix = "env LLVM_PROFILE_FILE=%s " % profilefile |
| context.profilefiles.append(profilefile) |
| return prefix + commandline |
| |
| |
| def _mutateScript(context, script): |
| return testplan.mutateScript(context, script, _mutateCommandline) |
| |
| |
| def mutatePlan(context, plan): |
| context.profilefiles = [] |
| # Adjust run steps to set LLVM_PROFILE_FILE environment variable. |
| plan.runscript = _mutateScript(context, plan.runscript) |
| plan.profile_files += context.profilefiles |
| |
| # Run profdata merge at the end |
| profdatafile = context.executable + ".profdata" |
| args = ['merge', '-output=%s' % profdatafile] + context.profilefiles |
| mergecmd = shellcommand.ShellCommand(context.config.llvm_profdata, args) |
| plan.profilescript += [mergecmd.toCommandline()] |
