add memory profiling using memray [skip ci]

NOEL-MNI · Jan 23, 2023 · 2b36b5b · 2b36b5b
1 parent 4aba9f8
commit 2b36b5b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -1,3 +1,5 @@
 # ignore deepMask
 # git clone <> in Dockerfile instead
-app/deepMask
+app/deepMask
+memray*.bin
+memray*.html
diff --git a/.gitignore b/.gitignore
@@ -2,4 +2,6 @@ data
 __pycache__
 notebooks/
 .ipynb_checkpoints/
-*.csv
+*.csv
+memray*.bin
+memray*.html
diff --git a/Makefile b/Makefile
@@ -18,6 +18,12 @@ clean-build:
 test-pipeline:
 	./app/inference.py $(CASE_ID) t1.nii.gz flair.nii.gz $(TMPDIR) cuda0 $(BRAIN_MASKING) $(PREPROCESS)
 
+memray-profiling:
+	python3 -m memray run ./app/inference.py $(CASE_ID) t1_brain.nii.gz t2_brain.nii.gz $(TMPDIR) cuda0 0 0
+
+memray-profiling-cpu:
+	python3 -m memray run ./app/inference.py $(CASE_ID) t1_brain.nii.gz t2_brain.nii.gz $(TMPDIR) cpu 0 0
+
 test-preprocess:
 	./app/preprocess.sh $(CASE_ID) t1.nii.gz flair.nii.gz $(TMPDIR) $(BRAIN_MASKING) $(PREPROCESS)