Fixed PyTorch caching and added ccache.

facebookresearch · Jul 17, 2019 · 4ec561a · 4ec561a
1 parent e1e28f3
commit 4ec561a
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -89,7 +89,7 @@ jobs:
                 export PATH=$HOME/miniconda/bin:$PATH
                 conda create -y -n habitat python=3.6
                 . activate habitat
-                conda install -q -y -c conda-forge ninja numpy pytest
+                conda install -q -y -c conda-forge ninja ccache numpy pytest
               fi
       - run:
           name: Install pytorch
@@ -107,6 +107,17 @@ jobs:
       - restore_cache:
           keys:
             - habitat-sim-{{ checksum "./hsim_sha" }}
+      - restore_cache:
+          keys:
+            - ccache-{{ arch }}-{{ .Branch }}-{{ epoch }}
+          paths:
+            - /home/circleci/.ccache
+      - run:
+          name: CCache initialization
+          command: |
+            ccache --show-stats
+            ccache --zero-stats
+            ccache --max-size=10.0G
       - run:
           name: Build, install habitat-sim and run benchmark
           no_output_timeout: 20m
@@ -125,6 +136,11 @@ jobs:
                   python setup.py install --headless
                 fi
               fi
+      - run:
+          name: ccache stats
+          when: always
+          command: |
+            ccache --show-stats
       - save_cache:
           key: conda-{{ checksum "habitat-api/.circleci/config.yml" }}
           background: true
@@ -152,6 +168,10 @@ jobs:
           key: habitat-sim-{{ checksum "./hsim_sha" }}
           paths:
             - ./habitat-sim
+      - save_cache:
+          key: ccache-{{ arch }}-{{ .Branch }}-{{ epoch }}
+          paths:
+            - /home/circleci/.ccache
       - run:
           name: Run api tests
           command: |