Skip to content

Commit

Permalink
fVDB CI debug free memory
Browse files Browse the repository at this point in the history
Signed-off-by: Jonathan Swartz <[email protected]>
  • Loading branch information
swahtz committed Sep 3, 2024
1 parent b58fe72 commit 27cb5d4
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion .github/workflows/fvdb.yml
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,9 @@ jobs:

- name: Build fvdb
run: |
cd fvdb
cd fvdb;
free -g;
free -g | awk '/^Mem:/{jobs=int($4); print "Free memory (GB): " jobs}';
DISTUTILS_DEBUG="1" TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6+PTX" MAX_JOBS=$(($(nproc) < $(free -g | awk '/^Mem:/{jobs=int($4/2.5); if(jobs<1) jobs=1; print jobs}') ? $(nproc) : $(free -g | awk '/^Mem:/{jobs=int($4/2.5); if(jobs<1) jobs=1; print jobs}'))) python setup.py bdist_wheel --dist-dir=dist
shell:
bash -el {0}
Expand Down

0 comments on commit 27cb5d4

Please sign in to comment.