Add initial profiling setup

flexflow · Feb 9, 2025 · 07b293d · 07b293d
1 parent 9d741dc
commit 07b293d
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 10 deletions.
diff --git 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
diff --git 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