1 files changed, 8 insertions, 2 deletions
diff --git a/tools/profile b/tools/profile
index 927467bc4e..0ccef6c26c 100755
--- a/tools/profile
+++ b/tools/profile
@@ -60,8 +60,14 @@ after = GC.allocated_size
 usage = (after - before) / 1024.0
 
 if mode
-  File.open("#{File.basename(path, '.rb')}.#{mode}.callgrind", 'w') do |out|
-    RubyProf::CallTreePrinter.new(results).print(out)
+  if RubyProf.const_defined?(:CallStackPrinter)
+    File.open("#{File.basename(path, '.rb')}.#{mode}.html", 'w') do |out|
+      RubyProf::CallStackPrinter.new(results).print(out)
+    end
+  else
+    File.open("#{File.basename(path, '.rb')}.#{mode}.callgrind", 'w') do |out|
+      RubyProf::CallTreePrinter.new(results).print(out)
+    end
   end
 end