my $ProfilePass = "-insert-edge-profiling";
my $LLVMProfOpts = "";
my $ProgramOpts = "";
my $ProfileFile = "";
while (scalar(@ARGV) and ($_ = $ARGV[0], /^[-+]/)) {
shift;
last if /^--$/;
if (/^-?-block$/) { $ProfilePass = "-insert-block-profiling"; next; }
if (/^-?-edge$/) { $ProfilePass = "-insert-edge-profiling"; next; }
if (/^-?-function$/) { $ProfilePass = "-insert-function-profiling"; next; }
if (/^-?-o$/) { die "-o option requires a filename argument!" if (!scalar(@ARGV));
$ProgramOpts .= " -llvmprof-output $ARGV[0]";
$ProfileFile = $ARGV[0];
shift;
next;
}
if (/^-?-help$/) {
print "OVERVIEW: profile.pl - Instrumentation and profile printer.\n\n";
print "USAGE: profile.pl [options] program.bc <program args>\n\n";
print "OPTIONS:\n";
print " -block - Enable basicblock profiling\n";
print " -edge - Enable edge profiling\n";
print " -function - Enable function profiling\n";
print " -o <file> - Specify an output file other than llvm-prof.out.\n";
print " -help - Print this usage information\n";
print "\nAll other options are passed into llvm-prof.\n";
exit 1;
}
$LLVMProfOpts .= " " . $_;
}
die "Must specify LLVM bitcode file as first argument!" if (@ARGV == 0);
my $BytecodeFile = $ARGV[0];
shift @ARGV;
my $libdir = `llvm-config --libdir`;
chomp $libdir;
my $LibProfPath = $libdir . "/libprofile_rt.so";
system "opt -q -f $ProfilePass $BytecodeFile -o $BytecodeFile.inst";
system "lli -fake-argv0 '$BytecodeFile' -load $LibProfPath " .
"$BytecodeFile.inst $ProgramOpts " . (join ' ', @ARGV);
system "rm $BytecodeFile.inst";
system "llvm-prof $LLVMProfOpts $BytecodeFile $ProfileFile";