From 07b293dba006cb2309f0634e13262f57fa82dec8 Mon Sep 17 00:00:00 2001
From: Colin Unger <lockshaw@lockshaw.net>
Date: Sat, 8 Feb 2025 20:59:23 -0800
Subject: [PATCH] Add initial profiling setup

---
 .proj.toml | 17 +++++++++++++++++
 flake.lock | 18 +++++++++---------
 flake.nix  |  4 +++-
 3 files changed, 29 insertions(+), 10 deletions(-)

diff --git a/.proj.toml b/.proj.toml
index 84aa321f17..413d628246 100644
--- a/.proj.toml
+++ b/.proj.toml
@@ -3,6 +3,23 @@ testsuite_macro = "FF_TEST_SUITE"
 namespace_name = "FlexFlow"
 header_extension = ".h"
 
+bin_targets = [
+  "export-model-arch",
+  "substitution-to-dot",
+]
+
+lib_targets = [
+  "utils",
+  "op-attrs",
+  "kernels",
+  "pcg",
+  "substitutions",
+  "compiler",
+  "substitution-generator",
+  "local-execution",
+  "models",
+]
+
 build_targets = [
   "utils",
   "op-attrs",
diff --git a/flake.lock b/flake.lock
index fd082340f1..5cbe8efe09 100644
--- a/flake.lock
+++ b/flake.lock
@@ -66,17 +66,17 @@
         ]
       },
       "locked": {
-        "lastModified": 1739072952,
-        "narHash": "sha256-8Y5Ljq2DtitTeYa8HvedBzugWTI1gSTK5steaEn5T/A=",
-        "owner": "lockshaw",
-        "repo": "proj",
-        "rev": "a48269850edfafb4df406ea956ffd9be3918190f",
-        "type": "github"
+        "lastModified": 1739076652,
+        "narHash": "sha256-gajL4+SoNE4iPUBeU0Ez6y1CRqi8fTot2v5SqppHjVQ=",
+        "ref": "refs/heads/master",
+        "rev": "6cee774e2ed14d659a87ab464dcf49ff6a0ed0f2",
+        "revCount": 70,
+        "type": "git",
+        "url": "file:///home/lockshaw/x/proj/proj"
       },
       "original": {
-        "owner": "lockshaw",
-        "repo": "proj",
-        "type": "github"
+        "type": "git",
+        "url": "file:///home/lockshaw/x/proj/proj"
       }
     },
     "root": {
diff --git a/flake.nix b/flake.nix
index 750a441f84..390fac2b96 100644
--- a/flake.nix
+++ b/flake.nix
@@ -18,7 +18,7 @@
     flake-utils.url = "github:numtide/flake-utils";
 
     proj-repo = {
-      url = "github:lockshaw/proj";
+      url = "git+file:///home/lockshaw/x/proj/proj";
       inputs.nixpkgs.follows = "nixpkgs";
       inputs.flake-utils.follows = "flake-utils";
     };
@@ -120,6 +120,7 @@
               lcov # for code coverage
               compdb
               gbenchmark
+              linuxPackages_latest.perf
             ])
             (with proj-repo.packages.${system}; [
               proj
@@ -162,6 +163,7 @@
               ruff
               jq
               gh
+              hotspot
             ])
             (with pkgs.python3Packages; [
               gitpython