Initial import. Support for findfaster.

Author: Sean Heelan

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length=120

.gitignore

@@ -0,0 +1,2 @@
+venv
+__pycache__/

.vscode/settings.json

@@ -0,0 +1,5 @@
+{
+    "python.linting.flake8Enabled": true,
+    "python.linting.pylintEnabled": false,
+    "python.linting.enabled": true
+}

perf-gpt.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+
+# perf-gpt is an experimental tool for performance analysis and optimisation
+# using LLMs. Its purpose is to take data from existing profilers and provide
+# the user with helpful summaries, advice and direction.
+#
+# Author: Sean Heelan
+# Email: sean@heelan.io
+
+from perfgpt.commands import (
+    findfaster,
+    topn
+)
+
+import argparse
+import os
+import sys
+
+import openai
+
+from dotenv import load_dotenv
+load_dotenv()
+
+openai.api_key = os.environ["OPENAI_API_KEY"]
+
+commands = {
+    findfaster.command: findfaster,
+    topn.command: topn
+}
+
+parser = argparse.ArgumentParser(
+    prog=sys.argv[0],
+    description="Performance analysis and optimisation with LLMs"
+)
+parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
+parser.add_argument("-d", "--debug", action="store_true", help="Debug output")
+parser.add_argument("--temperature", type=float, default=0, help="ChatGPT temperature. See OpenAI docs.")
+
+subparsers = parser.add_subparsers(help="sub-command help", dest="command")
+for v in commands.values():
+    v.add_to_command_parser(subparsers)
+
+args = parser.parse_args()
+
+if not args.command:
+    parser.print_help(sys.stderr)
+    sys.stderr.write("\nNo sub-command selected\n")
+    sys.exit(1)
+
+if args.command not in commands:
+    parser.print_help(sys.stderr)
+    sys.stderr.write("\nUnknown sub-command\n")
+    sys.exit(1)
+
+sys.exit(commands[args.command].run(parser, args))

perfgpt/commands/findfaster.py

@@ -0,0 +1,55 @@
+import openai
+
+command = "findfaster"
+help = "Try to find a faster version of the provided library or program"
+
+
+def add_to_command_parser(subparsers):
+    parser = subparsers.add_parser(command)
+    parser.add_argument("-t", "--software-type", choices=["program", "library", "pylibrary"], default="library",
+                        help="Specify the type of software. Not necessary, but can lead to better results.")
+    parser.add_argument("target", help="The program or library to find a faster version of")
+
+
+software_type_prompts = {
+    "program": """What are the fastest and most memory-efficient programs that provide
+                the same functionality as {target}, specifically those that use SIMD instructions
+                or are optimized for high performance? Output the suggested programs in a list.
+                For each program, give a summary.""",
+    "library": """What are the fastest and most memory-efficient libraries that provide
+                the same functionality as {target}, specifically those that use SIMD instructions
+                or are optimized for high performance? Output the suggested libraries in a list.
+                For each library, give a summary.""",
+    "pylibrary": """What are the fastest and most memory-efficient Python libraries that
+                provide the same functionality as Python's {target} library, specifically those that use
+                SIMD instructions or are optimized for high performance? I'm also interested in any
+                lightweight, low-level libraries that may provide better performance than pure-Python
+                libraries. Output the suggested libraries in a list. For each library, give a summary."""
+}
+
+
+def run(args_parser, args):
+    target = args.target
+    temp = args.temperature
+
+    prompt = software_type_prompts[args.software_type]
+
+    completion = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        temperature=temp,
+        messages=[
+            {
+                "role": "system",
+                "content": """You are perf-gpt, a helpful assistant for performance analysis and optimisation
+                of software. Answer as concisely as possible."""
+            },
+            {
+                "role": "user",
+                "content": prompt.format(target=target)
+            }
+        ]
+    )
+
+    result = completion['choices'][0]['message']['content']
+    print(result)
+    return 0

perfgpt/commands/topn.py

@@ -0,0 +1,7 @@
+command = "topn"
+help = "Analyse Top-N output from a profiler"
+
+
+def add_to_command_parser(subparsers):
+    parser = subparsers.add_parser(command)
+    parser.add_argument("-f", help="File containing the Top N data")