Enable cross-directory ccache

.flake/pkgs/ffdb/default.nix

@@ -0,0 +1,40 @@
+{ lib
+, stdenv
+, makeWrapper
+, gdb
+, python3
+, proj
+}:
+
+stdenv.mkDerivation rec {
+  pname = "ffdb";
+  version = "0.1";
+
+  pythonPath = with python3.pkgs; makePythonPath [
+    proj
+  ];
+
+  dontBuild = true;
+
+  nativeBuildInputs = [ makeWrapper ];
+
+  src = ./.;
+
+  installPhase = ''
+    mkdir -p $out/share/ffdb
+    cp ffdb.py $out/share/ffdb
+    makeWrapper ${gdb}/bin/gdb $out/bin/gdb \
+      --add-flags "-q -x $out/share/ffdb/ffdb.py" \
+      --set NIX_PYTHONPATH ${pythonPath} \
+      --prefix PATH : ${lib.makeBinPath [
+        python3
+      ]}
+    cp $out/bin/gdb $out/bin/ffdb
+  '';
+
+  nativeCheckInputs = [
+    gdb
+    python3
+    proj
+  ];
+}

.flake/pkgs/ffdb/ffdb.py

@@ -0,0 +1,112 @@
+from proj.config_file import get_config_root
+from pathlib import Path
+import gdb
+
+gdb.execute(f'directory {get_config_root(Path.cwd())}')
+gdb.prompt_hook = lambda x: '(ffdb) '
+gdb.execute('set history save on')
+
+# python
+# from proj.config_file import get_config_root
+# from pathlib import Path
+# print(Path.cwd())
+# #gdb.execute(f'directory {get_conf}/{}')
+# end
+
+# show directories
+
+# import gdb.printing
+#
+# class NodePrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         ptr = self.val["ptr"]
+#         if ptr != 0:
+#             op_type = ptr.referenced_value()['op_type']
+#             return f'Node<guid={self.val["guid"]} ptr={ptr} op_type={op_type}>'
+#         else:
+#             return f'Node<guid={self.val["guid"]} ptr={self.val["ptr"]}>'
+#
+# class EdgePrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         return f'Edge<src={self.val["srcOp"]["guid"]} dst={self.val["dstOp"]["guid"]}>'
+#
+# class MachineViewPrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         toks = []
+#         if self.val['device_type'] == 0:
+#             toks.append('type=GPU')
+#         else:
+#             toks.append('type=CPU')
+#         start_device_id = self.val['start_device_id']
+#         for i in range(self.val['ndims']):
+#             dim = self.val['dim'][i]
+#             stride = self.val['stride'][i]
+#             toks.append(f'{i}=[{start_device_id}:{start_device_id+dim}:{stride}]')
+#         return f'MachineView<{" ".join(toks)}>'
+#
+# class DomainPrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         toks = []
+#         ndim = self.val['dim']
+#         for i in range(ndim):
+#             lo = self.val['rect_data'][i]
+#             hi = self.val['rect_data'][i + ndim]
+#             toks.append(f'{i}=[{lo}:{hi}]')
+#         return f'Domain<{" ".join(toks)}>'
+#
+# class TensorShapePrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         toks = []
+#         ndim = self.val['num_dims']
+#         for i in range(ndim):
+#             dim = self.val['dims'][i]
+#             size = dim['size']
+#             degree = dim['degree']
+#             parallel_idx = dim['parallel_idx']
+#             toks.append(f'{i}=[s={size} d={degree} pi={parallel_idx}]')
+#         return f'TensorShape<{" ".join(toks)}>'
+#
+# class ParallelTensorBasePrinter:
+#     def __init__(self, val):
+#         self.val = val
+#
+#     def to_string(self):
+#         toks = []
+#         toks.append(f'guid={self.val["parallel_tensor_guid"]}')
+#         ndim = self.val['num_dims']
+#         for i in range(ndim):
+#             dim = self.val['dims'][i]
+#             size = dim['size']
+#             degree = dim['degree']
+#             parallel_idx = dim['parallel_idx']
+#             toks.append(f'{i}=[s={size} d={degree} pi={parallel_idx}]')
+#         return f'ParallelTensorBase<{" ".join(toks)}>'
+#
+# def build_pretty_printer():
+#     pp = gdb.printing.RegexpCollectionPrettyPrinter(
+#         "flexflow")
+#     pp.add_printer('Node', '^FlexFlow::PCG::Node$', NodePrinter)
+#     pp.add_printer('Edge', '^FlexFlow::PCG::Edge$', EdgePrinter)
+#     pp.add_printer('MachineView', '^FlexFlow::MachineView$', MachineViewPrinter)
+#     pp.add_printer('Domain', '^Legion::Domain$', DomainPrinter)
+#     pp.add_printer('ParallelTensorShape', '^FlexFlow::ParallelTensorShape$', TensorShapePrinter)
+#     pp.add_printer('ParallelTensorBase', '^FlexFlow::ParallelTensorBase$', ParallelTensorBasePrinter)
+#     return pp
+#
+# gdb.printing.register_pretty_printer(
+#         gdb.current_objfile(), build_pretty_printer(), replace=True)

.gitignore

@@ -1,3 +1,6 @@
+# gdb history
+.gdb_history
+
 # dtgen files
 *.dtg.cc
 *.dtg.h

cmake/flexflow-utils.cmake

@@ -39,7 +39,7 @@ function(ff_set_cxx_properties target)
       CXX_EXTENSIONS NO
   )
   target_compile_options(${target}
-    PRIVATE $<$<COMPILE_LANGUAGE:CXX>:> # add C++ compile flags here
+    PRIVATE $<$<COMPILE_LANGUAGE:CXX>:> "-ffile-prefix-map=${CMAKE_SOURCE_DIR}=." # add C++ compile flags here
   )
 endfunction()
 

flake.nix

@@ -35,10 +35,13 @@
       mkShell = pkgs.mkShell.override {
         stdenv = pkgs.cudaPackages.backendStdenv;
       };
+
+      proj = proj-repo.packages.${system}.proj;
     in 
     {
       packages = {
         legion = pkgs.callPackage ./.flake/pkgs/legion.nix { };
+        ffdb = pkgs.callPackage ./.flake/pkgs/ffdb { inherit proj; };
         hpp2plantuml = pkgs.python3Packages.callPackage ./.flake/pkgs/hpp2plantuml.nix { };
         rapidcheckFull = pkgs.symlinkJoin {
           name = "rapidcheckFull";
@@ -102,9 +105,7 @@
               doxygen
               lcov # for code coverage
             ])
-            (with proj-repo.packages.${system}; [
-              proj
-            ])
+            [ proj ]
             (with self.packages.${system}; [
               legion
               hpp2plantuml
@@ -128,7 +129,6 @@
               gh-markdown-preview
               shellcheck
               plantuml
-              gdb
               ruff
               compdb
               jq
@@ -148,6 +148,9 @@
               black
               toml
             ])
+            (with self.packages.${system}; [
+              ffdb
+            ])
           ];
         };
       };