Don't take the lock by unless needed at exit

theano/gof/cmodule.py

@@ -1299,13 +1299,17 @@ class ModuleCache(object):
         else:
             age_thresh_use = None
 
+        too_old_to_use = self.refresh(
+            age_thresh_use=age_thresh_use,
+            delete_if_problem=delete_if_problem,
+            # The clean up is done at init, no need to trigger it again
+            cleanup=False)
+        if not too_old_to_use:
+            return
         with compilelock.lock_ctx():
             # Update the age of modules that have been accessed by other
             # processes and get all module that are too old to use
             # (not loaded in self.entry_from_key).
-            too_old_to_use = self.refresh(
-                age_thresh_use=age_thresh_use,
-                delete_if_problem=delete_if_problem)
 
             for entry in too_old_to_use:
                 # TODO: we are assuming that modules that haven't been
@@ -1392,86 +1396,101 @@ class ModuleCache(object):
         if min_age is None:
             min_age = self.age_thresh_del_unversioned
 
-        with compilelock.lock_ctx():
-            all_key_datas = list(self.module_hash_to_key_data.values())
-            for key_data in all_key_datas:
-                if not key_data.keys:
-                    # May happen for broken versioned keys.
-                    continue
-                for key_idx, key in enumerate(key_data.keys):
-                    version, rest = key
-                    if version:
-                        # Since the version is included in the module hash,
-                        # it should not be possible to mix versioned and
-                        # unversioned keys in the same KeyData object.
-                        assert key_idx == 0
-                        break
-                if not version:
-                    # Note that unversioned keys cannot be broken, so we can
-                    # set do_manual_check to False to speed things up.
-                    key_data.delete_keys_from(self.entry_from_key,
-                                              do_manual_check=False)
-                    entry = key_data.get_entry()
-                    # Entry is guaranteed to be in this dictionary, because
-                    # an unversioned entry should never have been loaded via
-                    # refresh.
-                    assert entry in self.module_from_name
-
-                    del self.module_from_name[entry]
-                    del self.module_hash_to_key_data[key_data.module_hash]
+        # As this delete object that we build and other don't use, we
+        # don't need the lock.
+        all_key_datas = list(self.module_hash_to_key_data.values())
+        for key_data in all_key_datas:
+            if not key_data.keys:
+                # May happen for broken versioned keys.
+                continue
+            for key_idx, key in enumerate(key_data.keys):
+                version, rest = key
+                if version:
+                    # Since the version is included in the module hash,
+                    # it should not be possible to mix versioned and
+                    # unversioned keys in the same KeyData object.
+                    assert key_idx == 0
+                    break
+            if not version:
+                # Note that unversioned keys cannot be broken, so we can
+                # set do_manual_check to False to speed things up.
+                key_data.delete_keys_from(self.entry_from_key,
+                                          do_manual_check=False)
+                entry = key_data.get_entry()
+                # Entry is guaranteed to be in this dictionary, because
+                # an unversioned entry should never have been loaded via
+                # refresh.
+                assert entry in self.module_from_name
+
+                del self.module_from_name[entry]
+                del self.module_hash_to_key_data[key_data.module_hash]
 
-                    parent = os.path.dirname(entry)
-                    assert parent.startswith(os.path.join(self.dirname, 'tmp'))
-                    _rmtree(parent, msg='unversioned', level=logging.INFO,
-                            ignore_nocleanup=True)
+                parent = os.path.dirname(entry)
+                assert parent.startswith(os.path.join(self.dirname, 'tmp'))
+                _rmtree(parent, msg='unversioned', level=logging.INFO,
+                        ignore_nocleanup=True)
 
-            # Sanity check: all unversioned keys should have been removed at
-            # this point.
-            for key in self.entry_from_key:
-                assert key[0]
+        # Sanity check: all unversioned keys should have been removed at
+        # this point.
+        for key in self.entry_from_key:
+            assert key[0]
 
-            time_now = time.time()
-            for filename in os.listdir(self.dirname):
-                if filename.startswith('tmp'):
-                    try:
-                        open(os.path.join(self.dirname, filename, 'key.pkl')
-                             ).close()
-                        has_key = True
-                    except IOError:
-                        has_key = False
-                    if not has_key:
-                        # Use the compiled file by default
-                        path = module_name_from_dir(os.path.join(self.dirname,
-                                                                 filename),
-                                                    False)
-                        # If it don't exist, use any file in the directory.
-                        if path is None:
-                            path = os.path.join(self.dirname, filename)
-                            files = os.listdir(path)
-                            if files:
-                                path = os.path.join(path, files[0])
-                            else:
-                                # If the directory is empty skip it.
-                                # They are deleted elsewhere.
-                                continue
-                        age = time_now - last_access_time(path)
-
-                        # In normal case, the processus that created this
-                        # directory will delete it. However, if this processus
-                        # crashes, it will not be cleaned up.
-                        # As we don't know if this directory is still used,
-                        # we wait one week and suppose that the processus
-                        # crashed, and we take care of the clean-up.
-                        if age > min_age:
-                            _rmtree(os.path.join(self.dirname, filename),
-                                    msg='old unversioned', level=logging.INFO,
-                                    ignore_nocleanup=True)
+        to_del = []
+        time_now = time.time()
+        for filename in os.listdir(self.dirname):
+            if filename.startswith('tmp'):
+                try:
+                    open(os.path.join(self.dirname, filename, 'key.pkl')
+                    ).close()
+                    has_key = True
+                except IOError:
+                    has_key = False
+                if not has_key:
+                    # Use the compiled file by default
+                    path = module_name_from_dir(os.path.join(self.dirname,
+                                                             filename),
+                                                False)
+                    # If it don't exist, use any file in the directory.
+                    if path is None:
+                        path = os.path.join(self.dirname, filename)
+                        files = os.listdir(path)
+                        if files:
+                            path = os.path.join(path, files[0])
+                        else:
+                            # If the directory is empty skip it.
+                            # They are deleted elsewhere.
+                            continue
+                    age = time_now - last_access_time(path)
+
+                    # In normal case, the processus that created this
+                    # directory will delete it. However, if this processus
+                    # crashes, it will not be cleaned up.
+                    # As we don't know if this directory is still used,
+                    # we wait one week and suppose that the processus
+                    # crashed, and we take care of the clean-up.
+                    if age > min_age:
+                        to_del.append(os.path.join(self.dirname, filename))
+
+        if not to_del:
+            return
+        with compilelock.lock_ctx():
+            for f in to_del:
+                _rmtree(f,
+                    msg='old unversioned', level=logging.INFO,
+                    ignore_nocleanup=True)
 
     def _on_atexit(self):
         # Note: no need to call refresh() since it is called by clear_old().
-        with compilelock.lock_ctx():
-            self.clear_old()
-            self.clear_unversioned()
+
+        # Note: no need to take the lock. For unversioned files, we
+        # don't need it as they aren't shared. For old unversioned
+        # files, this happen rarely, so we take the lock only when
+        # this happen.
+
+        # Note: for clear_old(), as this happen unfrequently, we only
+        # take the lock when it happen.
+        self.clear_old()
+        self.clear_unversioned()
         _logger.debug('Time spent checking keys: %s',
                       self.time_spent_in_check_key)