X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=utils%2Fprofile.pl;h=f9950f97fea82179c1e22a23b820db1ef18211ac;hb=fff916a96076e284edb043d7541cfc7902086039;hp=27bfd7190cdb71ecce8b689ba0bac87fb0b355c6;hpb=ad910ebe70644040701f8f39fd96eed78ca2ed23;p=oota-llvm.git

diff --git a/utils/profile.pl b/utils/profile.pl
index 27bfd7190cd..f9950f97fea 100755
--- a/utils/profile.pl
+++ b/utils/profile.pl
@@ -8,12 +8,20 @@
 # Syntax:   profile.pl [OPTIONS] bytecodefile <arguments>
 #
 # OPTIONS may include one or more of the following:
-#     NONE SO FAR
+#     -block    - Enable basicblock profiling
+#     -edge     - Enable edge profiling
+#     -function - Enable function profiling
+#     -o <filename> - Emit profiling information to the specified file, instead
+#                     of llvmprof.out
 #
+# Any unrecognized options are passed into the invocation of llvm-prof
 #
 
+my $ProfilePass = "-insert-edge-profiling";
 
-my $ProfilePass = "-insert-function-profiling";
+my $LLVMProfOpts = "";
+my $ProgramOpts = "";
+my $ProfileFile = "";
 
 # Parse arguments...
 while (scalar(@ARGV) and ($_ = $ARGV[0], /^[-+]/)) {
@@ -21,9 +29,31 @@ while (scalar(@ARGV) and ($_ = $ARGV[0], /^[-+]/)) {
   last if /^--$/;  # Stop processing arguments on --
 
   # List command line options here...
-  #if (/^-enable-foo$/)     { $FOO = 1; next; }
+  if (/^-?-block$/)    { $ProfilePass = "-insert-block-profiling"; next; }
+  if (/^-?-edge$/)     { $ProfilePass = "-insert-edge-profiling"; next; }
+  if (/^-?-function$/) { $ProfilePass = "-insert-function-profiling"; next; }
+  if (/^-?-o$/) {         # Read -o filename...
+    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;
+  }
 
-  print "Unknown option: $_ : ignoring!\n";
+  # Otherwise, pass the option on to llvm-prof
+  $LLVMProfOpts .= " " . $_;
 }
 
 die "Must specify LLVM bytecode file as first argument!" if (@ARGV == 0);
@@ -32,13 +62,13 @@ my $BytecodeFile = $ARGV[0];
 
 shift @ARGV;
 
-my $LLIPath = `which lli`;
-$LLIPath = `dirname $LLIPath`;
-chomp $LLIPath;
+my $libdir = `llvm-config --libdir`;
+chomp $libdir;
 
-my $LibProfPath = $LLIPath . "/../../lib/Debug/libprofile_rt.so";
+my $LibProfPath = $libdir . "/profile_rt.so";
 
-system "opt $ProfilePass < $BytecodeFile | lli -load $LibProfPath - " .
-         (join ' ', @ARGV);
-
-system "llvm-prof $BytecodeFile";
+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";