Merge pull request #380 from opesci/dask2

Added draft of devito dask FWI tutorial

devito/compiler.py

@@ -1,7 +1,7 @@
 from functools import partial
 from hashlib import sha1
-from os import environ, getuid, mkdir, path
-from tempfile import gettempdir
+from os import environ, path
+from tempfile import mkdtemp
 from time import time
 from sys import platform
 import subprocess
@@ -194,12 +194,13 @@ def get_tmp_dir():
 
     :return: Path to a devito-specific tmp directory
     """
-    tmpdir = path.join(gettempdir(), "devito-%s" % getuid())
+    global _devito_compiler_tmpdir
+    try:
+        path.exists(_devito_compiler_tmpdir)
+    except:
+        _devito_compiler_tmpdir = mkdtemp(prefix="devito-")
 
-    if not path.exists(tmpdir):
-        mkdir(tmpdir)
-
-    return tmpdir
+    return _devito_compiler_tmpdir
 
 
 def load(basename, compiler):

docs/tutorials.rst

@@ -56,3 +56,5 @@ inversion problems.
   <http://nbviewer.jupyter.org/github/opesci/devito/blob/master/examples/seismic/tutorials/02_rtm.ipynb>`_
 * `03 - Full-Waveform Inversion (FWI)
   <http://nbviewer.jupyter.org/github/opesci/devito/blob/master/examples/seismic/tutorials/03_fwi.ipynb>`_
+* `04 - Parallel FWI with Dask
+  <http://nbviewer.jupyter.org/github/opesci/devito/blob/master/examples/seismic/tutorials/04_dask.ipynb>`_

environment.yml

@@ -16,6 +16,9 @@ dependencies:
   - psutil>=5.1.0
   - sphinx
   - sphinx_rtd_theme
+  - dask
+  - distributed
+  - scikit-image
   - pip:
     - "git+https://github.com/inducer/codepy"
     - py-cpuinfo

requirements.txt

@@ -10,3 +10,6 @@ psutil>=5.1.0
 py-cpuinfo
 git+https://github.com/inducer/cgen
 git+https://github.com/inducer/codepy
+dask
+distributed
+scikit-image