launch multiprocessing programs without in python mode

pythonprofilers · Mar 21, 2017 · 52c5788 · 52c5788
1 parent d333fda
commit 52c5788
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/examples/multiprocessing_example.py b/examples/multiprocessing_example.py
@@ -2,7 +2,7 @@
 An undecorated example of a script that allocates memory in multiprocessing
 workers to demonstrate the use of memory_profiler with multiple processes.
 
-Run this script with mprof run -M python multiprocessing_example.py
+Run this script with mprof run -M multiprocessing_example.py
 You can then visualize the usage with mprof plot.
 """
 

diff --git a/mprof b/mprof
@@ -219,7 +219,15 @@ def run_action():
 
     # .. TODO: more than one script as argument ? ..
     if args[0].endswith('.py') and not options.nopython:
-        options.python = True
+        if options.multiprocess and not args[0].startswith("python"):
+            # in multiprocessing mode you want to spawn a separate
+            # python process
+            args.insert(0, "python")
+        else:
+            options.python = True
+    if options.multiprocess:
+        # multiprocess can't work in python mode
+        options.python = False
     if options.python:
         print("running as a Python program...")
         if not args[0].startswith("python"):