2016-03-28 17:37:34 -04:00
|
|
|
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
2017-09-07 16:11:58 +02:00
|
|
|
use rustc_data_structures::stable_hasher::{HashStable, StableHasher,
|
|
|
|
StableHashingContextProvider};
|
2017-08-21 16:44:05 +02:00
|
|
|
use rustc_data_structures::fx::{FxHashMap, FxHashSet};
|
|
|
|
use rustc_data_structures::indexed_vec::{Idx, IndexVec};
|
2016-07-25 10:51:14 -04:00
|
|
|
use session::config::OutputType;
|
2016-07-21 12:33:23 -04:00
|
|
|
use std::cell::{Ref, RefCell};
|
2017-08-21 16:44:05 +02:00
|
|
|
use std::hash::Hash;
|
2016-03-28 17:37:34 -04:00
|
|
|
use std::rc::Rc;
|
2017-09-25 13:51:49 +02:00
|
|
|
use ty::TyCtxt;
|
2017-07-23 10:02:07 -06:00
|
|
|
use util::common::{ProfileQueriesMsg, profq_msg};
|
2016-03-28 17:37:34 -04:00
|
|
|
|
2017-09-07 16:11:58 +02:00
|
|
|
use ich::Fingerprint;
|
|
|
|
|
2017-06-23 16:37:12 +02:00
|
|
|
use super::dep_node::{DepNode, DepKind, WorkProductId};
|
2016-03-28 17:37:34 -04:00
|
|
|
use super::query::DepGraphQuery;
|
|
|
|
use super::raii;
|
isolate dep-graph tasks
A task function is now given as a `fn` pointer to ensure that it carries
no state. Each fn can take two arguments, because that worked out to be
convenient -- these two arguments must be of some type that is
`DepGraphSafe`, a new trait that is intended to prevent "leaking"
information into the task that was derived from tracked state.
This intentionally leaves `DepGraph::in_task()`, the more common form,
alone. Eventually all uses of `DepGraph::in_task()` should be ported
to `with_task()`, but I wanted to start with a smaller subset.
Originally I wanted to use closures bound by an auto trait, but that
approach has some limitations:
- the trait cannot have a `read()` method; since the current method
is unused, that may not be a problem.
- more importantly, we would want the auto trait to be "undefined" for all types
*by default* -- that is, this use case doesn't really fit the typical
auto trait scenario. For example, imagine that there is a `u32` loaded
out of a `hir::Node` -- we don't really want to be passing that
`u32` into the task!
2017-03-06 15:35:34 -05:00
|
|
|
use super::safe::DepGraphSafe;
|
2017-08-21 16:44:05 +02:00
|
|
|
use super::edges::{self, DepGraphEdges};
|
2017-09-22 13:00:42 +02:00
|
|
|
use super::serialized::{SerializedDepGraph, SerializedDepNodeIndex};
|
|
|
|
use super::prev::PreviousDepGraph;
|
2016-03-28 17:37:34 -04:00
|
|
|
|
2017-09-25 13:51:49 +02:00
|
|
|
|
2016-03-28 17:37:34 -04:00
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct DepGraph {
|
2017-09-14 17:43:03 +02:00
|
|
|
data: Option<Rc<DepGraphData>>,
|
2017-09-19 12:13:09 +02:00
|
|
|
|
|
|
|
// At the moment we are using DepNode as key here. In the future it might
|
|
|
|
// be possible to use an IndexVec<DepNodeIndex, _> here. At the moment there
|
|
|
|
// are a few problems with that:
|
|
|
|
// - Some fingerprints are needed even if incr. comp. is disabled -- yet
|
|
|
|
// we need to have a dep-graph to generate DepNodeIndices.
|
|
|
|
// - The architecture is still in flux and it's not clear what how to best
|
|
|
|
// implement things.
|
2017-09-14 17:43:03 +02:00
|
|
|
fingerprints: Rc<RefCell<FxHashMap<DepNode, Fingerprint>>>
|
2016-07-21 12:33:23 -04:00
|
|
|
}
|
|
|
|
|
2017-08-21 16:44:05 +02:00
|
|
|
/// As a temporary measure, while transitioning to the new DepGraph
|
|
|
|
/// implementation, we maintain the old and the new dep-graph encoding in
|
|
|
|
/// parallel, so a DepNodeIndex actually contains two indices, one for each
|
|
|
|
/// version.
|
|
|
|
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
|
|
|
|
pub struct DepNodeIndex {
|
|
|
|
legacy: edges::DepNodeIndex,
|
|
|
|
new: DepNodeIndexNew,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl DepNodeIndex {
|
|
|
|
pub const INVALID: DepNodeIndex = DepNodeIndex {
|
|
|
|
legacy: edges::DepNodeIndex::INVALID,
|
|
|
|
new: DepNodeIndexNew::INVALID,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
|
|
|
|
pub enum DepNodeColor {
|
|
|
|
Red,
|
2017-09-25 13:51:49 +02:00
|
|
|
Green(DepNodeIndex)
|
2017-09-25 12:25:41 +02:00
|
|
|
}
|
|
|
|
|
2016-07-21 12:33:23 -04:00
|
|
|
struct DepGraphData {
|
2017-08-21 16:44:05 +02:00
|
|
|
/// The old, initial encoding of the dependency graph. This will soon go
|
|
|
|
/// away.
|
2017-07-04 15:06:57 +02:00
|
|
|
edges: RefCell<DepGraphEdges>,
|
2016-07-21 12:33:23 -04:00
|
|
|
|
2017-08-21 16:44:05 +02:00
|
|
|
/// The new encoding of the dependency graph, optimized for red/green
|
|
|
|
/// tracking. The `current` field is the dependency graph of only the
|
|
|
|
/// current compilation session: We don't merge the previous dep-graph into
|
|
|
|
/// current one anymore.
|
|
|
|
current: RefCell<CurrentDepGraph>,
|
|
|
|
|
2017-09-22 13:00:42 +02:00
|
|
|
/// The dep-graph from the previous compilation session. It contains all
|
|
|
|
/// nodes and edges as well as all fingerprints of nodes that have them.
|
|
|
|
previous: PreviousDepGraph,
|
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
colors: RefCell<FxHashMap<DepNode, DepNodeColor>>,
|
|
|
|
|
2016-07-22 10:39:30 -04:00
|
|
|
/// When we load, there may be `.o` files, cached mir, or other such
|
2016-07-21 12:33:23 -04:00
|
|
|
/// things available to us. If we find that they are not dirty, we
|
|
|
|
/// load the path to the file storing those work-products here into
|
|
|
|
/// this map. We can later look for and extract that data.
|
2017-06-06 15:09:21 +02:00
|
|
|
previous_work_products: RefCell<FxHashMap<WorkProductId, WorkProduct>>,
|
2016-07-21 12:33:23 -04:00
|
|
|
|
2016-07-22 10:39:30 -04:00
|
|
|
/// Work-products that we generate in this run.
|
2017-06-06 15:09:21 +02:00
|
|
|
work_products: RefCell<FxHashMap<WorkProductId, WorkProduct>>,
|
2017-06-12 17:00:55 +02:00
|
|
|
|
|
|
|
dep_node_debug: RefCell<FxHashMap<DepNode, String>>,
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
impl DepGraph {
|
2017-09-22 13:00:42 +02:00
|
|
|
|
|
|
|
pub fn new(prev_graph: PreviousDepGraph) -> DepGraph {
|
2016-03-28 17:37:34 -04:00
|
|
|
DepGraph {
|
2017-09-22 13:00:42 +02:00
|
|
|
data: Some(Rc::new(DepGraphData {
|
|
|
|
previous_work_products: RefCell::new(FxHashMap()),
|
|
|
|
work_products: RefCell::new(FxHashMap()),
|
|
|
|
edges: RefCell::new(DepGraphEdges::new()),
|
|
|
|
dep_node_debug: RefCell::new(FxHashMap()),
|
|
|
|
current: RefCell::new(CurrentDepGraph::new()),
|
|
|
|
previous: prev_graph,
|
2017-09-25 12:25:41 +02:00
|
|
|
colors: RefCell::new(FxHashMap()),
|
2017-09-22 13:00:42 +02:00
|
|
|
})),
|
|
|
|
fingerprints: Rc::new(RefCell::new(FxHashMap())),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn new_disabled() -> DepGraph {
|
|
|
|
DepGraph {
|
|
|
|
data: None,
|
2017-09-14 17:43:03 +02:00
|
|
|
fingerprints: Rc::new(RefCell::new(FxHashMap())),
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-12-20 22:46:11 +02:00
|
|
|
/// True if we are actually building the full dep-graph.
|
|
|
|
#[inline]
|
|
|
|
pub fn is_fully_enabled(&self) -> bool {
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data.is_some()
|
2016-12-20 22:46:11 +02:00
|
|
|
}
|
|
|
|
|
2017-06-02 17:36:30 +02:00
|
|
|
pub fn query(&self) -> DepGraphQuery {
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data.as_ref().unwrap().edges.borrow().query()
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
|
2016-10-18 14:46:41 +11:00
|
|
|
pub fn in_ignore<'graph>(&'graph self) -> Option<raii::IgnoreTask<'graph>> {
|
2017-08-21 16:44:05 +02:00
|
|
|
self.data.as_ref().map(|data| raii::IgnoreTask::new(&data.edges,
|
|
|
|
&data.current))
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn with_ignore<OP,R>(&self, op: OP) -> R
|
|
|
|
where OP: FnOnce() -> R
|
|
|
|
{
|
|
|
|
let _task = self.in_ignore();
|
|
|
|
op()
|
|
|
|
}
|
|
|
|
|
2017-03-08 09:14:27 -05:00
|
|
|
/// Starts a new dep-graph task. Dep-graph tasks are specified
|
|
|
|
/// using a free function (`task`) and **not** a closure -- this
|
|
|
|
/// is intentional because we want to exercise tight control over
|
|
|
|
/// what state they have access to. In particular, we want to
|
|
|
|
/// prevent implicit 'leaks' of tracked state into the task (which
|
|
|
|
/// could then be read without generating correct edges in the
|
|
|
|
/// dep-graph -- see the [README] for more details on the
|
|
|
|
/// dep-graph). To this end, the task function gets exactly two
|
|
|
|
/// pieces of state: the context `cx` and an argument `arg`. Both
|
|
|
|
/// of these bits of state must be of some type that implements
|
|
|
|
/// `DepGraphSafe` and hence does not leak.
|
|
|
|
///
|
|
|
|
/// The choice of two arguments is not fundamental. One argument
|
|
|
|
/// would work just as well, since multiple values can be
|
|
|
|
/// collected using tuples. However, using two arguments works out
|
|
|
|
/// to be quite convenient, since it is common to need a context
|
|
|
|
/// (`cx`) and some argument (e.g., a `DefId` identifying what
|
|
|
|
/// item to process).
|
|
|
|
///
|
|
|
|
/// For cases where you need some other number of arguments:
|
|
|
|
///
|
|
|
|
/// - If you only need one argument, just use `()` for the `arg`
|
|
|
|
/// parameter.
|
|
|
|
/// - If you need 3+ arguments, use a tuple for the
|
|
|
|
/// `arg` parameter.
|
|
|
|
///
|
|
|
|
/// [README]: README.md
|
2017-09-07 16:11:58 +02:00
|
|
|
pub fn with_task<C, A, R, HCX>(&self,
|
|
|
|
key: DepNode,
|
|
|
|
cx: C,
|
|
|
|
arg: A,
|
|
|
|
task: fn(C, A) -> R)
|
|
|
|
-> (R, DepNodeIndex)
|
|
|
|
where C: DepGraphSafe + StableHashingContextProvider<ContextType=HCX>,
|
|
|
|
R: HashStable<HCX>,
|
2016-03-28 17:37:34 -04:00
|
|
|
{
|
2017-07-04 17:33:43 +02:00
|
|
|
if let Some(ref data) = self.data {
|
2017-09-25 13:51:49 +02:00
|
|
|
debug_assert!(!data.colors.borrow().contains_key(&key));
|
|
|
|
|
2017-07-04 17:33:43 +02:00
|
|
|
data.edges.borrow_mut().push_task(key);
|
2017-08-21 16:44:05 +02:00
|
|
|
data.current.borrow_mut().push_task(key);
|
2017-07-23 10:02:07 -06:00
|
|
|
if cfg!(debug_assertions) {
|
|
|
|
profq_msg(ProfileQueriesMsg::TaskBegin(key.clone()))
|
|
|
|
};
|
2017-09-07 16:11:58 +02:00
|
|
|
|
|
|
|
// In incremental mode, hash the result of the task. We don't
|
|
|
|
// do anything with the hash yet, but we are computing it
|
|
|
|
// anyway so that
|
|
|
|
// - we make sure that the infrastructure works and
|
|
|
|
// - we can get an idea of the runtime cost.
|
|
|
|
let mut hcx = cx.create_stable_hashing_context();
|
|
|
|
|
2017-07-04 17:33:43 +02:00
|
|
|
let result = task(cx, arg);
|
2017-07-23 10:02:07 -06:00
|
|
|
if cfg!(debug_assertions) {
|
|
|
|
profq_msg(ProfileQueriesMsg::TaskEnd)
|
|
|
|
};
|
2017-08-21 16:44:05 +02:00
|
|
|
|
|
|
|
let dep_node_index_legacy = data.edges.borrow_mut().pop_task(key);
|
|
|
|
let dep_node_index_new = data.current.borrow_mut().pop_task(key);
|
2017-09-07 16:11:58 +02:00
|
|
|
|
|
|
|
let mut stable_hasher = StableHasher::new();
|
|
|
|
result.hash_stable(&mut hcx, &mut stable_hasher);
|
2017-09-14 17:43:03 +02:00
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
let current_fingerprint = stable_hasher.finish();
|
|
|
|
|
2017-09-14 17:43:03 +02:00
|
|
|
assert!(self.fingerprints
|
|
|
|
.borrow_mut()
|
2017-09-25 12:25:41 +02:00
|
|
|
.insert(key, current_fingerprint)
|
2017-09-14 17:43:03 +02:00
|
|
|
.is_none());
|
2017-09-07 16:11:58 +02:00
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
let prev_fingerprint = data.previous.fingerprint_of(&key);
|
|
|
|
|
|
|
|
let color = if Some(current_fingerprint) == prev_fingerprint {
|
2017-09-25 13:51:49 +02:00
|
|
|
DepNodeColor::Green(DepNodeIndex {
|
|
|
|
legacy: dep_node_index_legacy,
|
|
|
|
new: dep_node_index_new,
|
|
|
|
})
|
2017-09-25 12:25:41 +02:00
|
|
|
} else {
|
|
|
|
DepNodeColor::Red
|
|
|
|
};
|
|
|
|
|
|
|
|
assert!(data.colors.borrow_mut().insert(key, color).is_none());
|
|
|
|
|
2017-08-21 16:44:05 +02:00
|
|
|
(result, DepNodeIndex {
|
|
|
|
legacy: dep_node_index_legacy,
|
|
|
|
new: dep_node_index_new,
|
|
|
|
})
|
2017-07-04 17:33:43 +02:00
|
|
|
} else {
|
2017-09-14 17:43:03 +02:00
|
|
|
if key.kind.fingerprint_needed_for_crate_hash() {
|
|
|
|
let mut hcx = cx.create_stable_hashing_context();
|
|
|
|
let result = task(cx, arg);
|
|
|
|
let mut stable_hasher = StableHasher::new();
|
|
|
|
result.hash_stable(&mut hcx, &mut stable_hasher);
|
|
|
|
assert!(self.fingerprints
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(key, stable_hasher.finish())
|
|
|
|
.is_none());
|
|
|
|
(result, DepNodeIndex::INVALID)
|
|
|
|
} else {
|
|
|
|
(task(cx, arg), DepNodeIndex::INVALID)
|
|
|
|
}
|
2017-07-04 17:33:43 +02:00
|
|
|
}
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
|
2017-06-23 16:37:12 +02:00
|
|
|
/// Execute something within an "anonymous" task, that is, a task the
|
|
|
|
/// DepNode of which is determined by the list of inputs it read from.
|
2017-07-04 17:33:43 +02:00
|
|
|
pub fn with_anon_task<OP,R>(&self, dep_kind: DepKind, op: OP) -> (R, DepNodeIndex)
|
2017-06-23 16:37:12 +02:00
|
|
|
where OP: FnOnce() -> R
|
|
|
|
{
|
|
|
|
if let Some(ref data) = self.data {
|
|
|
|
data.edges.borrow_mut().push_anon_task();
|
2017-08-21 16:44:05 +02:00
|
|
|
data.current.borrow_mut().push_anon_task();
|
2017-06-23 16:37:12 +02:00
|
|
|
let result = op();
|
2017-08-21 16:44:05 +02:00
|
|
|
let dep_node_index_legacy = data.edges.borrow_mut().pop_anon_task(dep_kind);
|
2017-09-25 12:25:41 +02:00
|
|
|
let (new_dep_node, dep_node_index_new) = data.current
|
|
|
|
.borrow_mut()
|
|
|
|
.pop_anon_task(dep_kind);
|
|
|
|
if let Some(new_dep_node) = new_dep_node {
|
|
|
|
assert!(data.colors
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(new_dep_node, DepNodeColor::Red)
|
|
|
|
.is_none());
|
|
|
|
}
|
|
|
|
|
2017-08-21 16:44:05 +02:00
|
|
|
(result, DepNodeIndex {
|
|
|
|
legacy: dep_node_index_legacy,
|
|
|
|
new: dep_node_index_new,
|
|
|
|
})
|
2017-06-23 16:37:12 +02:00
|
|
|
} else {
|
2017-07-04 17:33:43 +02:00
|
|
|
(op(), DepNodeIndex::INVALID)
|
2017-06-23 16:37:12 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-07-04 15:06:57 +02:00
|
|
|
#[inline]
|
2017-06-02 17:36:30 +02:00
|
|
|
pub fn read(&self, v: DepNode) {
|
2017-07-04 15:06:57 +02:00
|
|
|
if let Some(ref data) = self.data {
|
|
|
|
data.edges.borrow_mut().read(v);
|
2017-09-22 13:00:42 +02:00
|
|
|
|
|
|
|
let mut current = data.current.borrow_mut();
|
|
|
|
if let Some(&dep_node_index_new) = current.node_to_node_index.get(&v) {
|
|
|
|
current.read_index(dep_node_index_new);
|
|
|
|
} else {
|
|
|
|
bug!("DepKind {:?} should be pre-allocated but isn't.", v.kind)
|
|
|
|
}
|
2016-10-18 14:46:41 +11:00
|
|
|
}
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
|
|
|
|
2017-07-04 17:33:43 +02:00
|
|
|
#[inline]
|
|
|
|
pub fn read_index(&self, v: DepNodeIndex) {
|
|
|
|
if let Some(ref data) = self.data {
|
2017-08-21 16:44:05 +02:00
|
|
|
data.edges.borrow_mut().read_index(v.legacy);
|
|
|
|
data.current.borrow_mut().read_index(v.new);
|
2017-07-04 17:33:43 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-07-04 15:06:57 +02:00
|
|
|
/// Only to be used during graph loading
|
|
|
|
#[inline]
|
|
|
|
pub fn add_edge_directly(&self, source: DepNode, target: DepNode) {
|
|
|
|
self.data.as_ref().unwrap().edges.borrow_mut().add_edge(source, target);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Only to be used during graph loading
|
|
|
|
pub fn add_node_directly(&self, node: DepNode) {
|
|
|
|
self.data.as_ref().unwrap().edges.borrow_mut().add_node(node);
|
|
|
|
}
|
|
|
|
|
2017-09-22 13:00:42 +02:00
|
|
|
pub fn fingerprint_of(&self, dep_node: &DepNode) -> Fingerprint {
|
|
|
|
self.fingerprints.borrow()[dep_node]
|
2017-08-18 20:24:19 +02:00
|
|
|
}
|
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
pub fn prev_fingerprint_of(&self, dep_node: &DepNode) -> Option<Fingerprint> {
|
2017-09-22 13:00:42 +02:00
|
|
|
self.data.as_ref().unwrap().previous.fingerprint_of(dep_node)
|
2017-09-14 17:43:03 +02:00
|
|
|
}
|
|
|
|
|
2016-07-21 12:33:23 -04:00
|
|
|
/// Indicates that a previous work product exists for `v`. This is
|
|
|
|
/// invoked during initial start-up based on what nodes are clean
|
|
|
|
/// (and what files exist in the incr. directory).
|
2017-06-06 15:09:21 +02:00
|
|
|
pub fn insert_previous_work_product(&self, v: &WorkProductId, data: WorkProduct) {
|
2016-07-21 12:33:23 -04:00
|
|
|
debug!("insert_previous_work_product({:?}, {:?})", v, data);
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data
|
|
|
|
.as_ref()
|
|
|
|
.unwrap()
|
|
|
|
.previous_work_products
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(v.clone(), data);
|
2016-07-21 12:33:23 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Indicates that we created the given work-product in this run
|
|
|
|
/// for `v`. This record will be preserved and loaded in the next
|
|
|
|
/// run.
|
2017-06-06 15:09:21 +02:00
|
|
|
pub fn insert_work_product(&self, v: &WorkProductId, data: WorkProduct) {
|
2016-07-21 12:33:23 -04:00
|
|
|
debug!("insert_work_product({:?}, {:?})", v, data);
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data
|
|
|
|
.as_ref()
|
|
|
|
.unwrap()
|
|
|
|
.work_products
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(v.clone(), data);
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
2016-07-21 12:33:23 -04:00
|
|
|
|
|
|
|
/// Check whether a previous work product exists for `v` and, if
|
|
|
|
/// so, return the path that leads to it. Used to skip doing work.
|
2017-06-06 15:09:21 +02:00
|
|
|
pub fn previous_work_product(&self, v: &WorkProductId) -> Option<WorkProduct> {
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data
|
|
|
|
.as_ref()
|
|
|
|
.and_then(|data| {
|
|
|
|
data.previous_work_products.borrow().get(v).cloned()
|
|
|
|
})
|
2016-07-21 12:33:23 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Access the map of work-products created during this run. Only
|
|
|
|
/// used during saving of the dep-graph.
|
2017-06-06 15:09:21 +02:00
|
|
|
pub fn work_products(&self) -> Ref<FxHashMap<WorkProductId, WorkProduct>> {
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data.as_ref().unwrap().work_products.borrow()
|
2016-07-21 12:33:23 -04:00
|
|
|
}
|
2017-01-16 17:54:20 -05:00
|
|
|
|
|
|
|
/// Access the map of work-products created during the cached run. Only
|
|
|
|
/// used during saving of the dep-graph.
|
2017-06-06 15:09:21 +02:00
|
|
|
pub fn previous_work_products(&self) -> Ref<FxHashMap<WorkProductId, WorkProduct>> {
|
2017-07-04 15:06:57 +02:00
|
|
|
self.data.as_ref().unwrap().previous_work_products.borrow()
|
2017-01-16 17:54:20 -05:00
|
|
|
}
|
2017-06-12 17:00:55 +02:00
|
|
|
|
|
|
|
#[inline(always)]
|
2017-06-23 16:37:12 +02:00
|
|
|
pub fn register_dep_node_debug_str<F>(&self,
|
|
|
|
dep_node: DepNode,
|
|
|
|
debug_str_gen: F)
|
2017-06-12 17:00:55 +02:00
|
|
|
where F: FnOnce() -> String
|
|
|
|
{
|
2017-08-30 11:53:57 -07:00
|
|
|
let dep_node_debug = &self.data.as_ref().unwrap().dep_node_debug;
|
2017-06-12 17:00:55 +02:00
|
|
|
|
2017-08-30 11:53:57 -07:00
|
|
|
if dep_node_debug.borrow().contains_key(&dep_node) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
let debug_str = debug_str_gen();
|
|
|
|
dep_node_debug.borrow_mut().insert(dep_node, debug_str);
|
2017-06-12 17:00:55 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
pub(super) fn dep_node_debug_str(&self, dep_node: DepNode) -> Option<String> {
|
2017-07-10 15:45:36 -04:00
|
|
|
self.data.as_ref().and_then(|t| t.dep_node_debug.borrow().get(&dep_node).cloned())
|
2017-06-12 17:00:55 +02:00
|
|
|
}
|
2017-09-22 13:00:42 +02:00
|
|
|
|
|
|
|
pub fn serialize(&self) -> SerializedDepGraph {
|
|
|
|
let fingerprints = self.fingerprints.borrow();
|
|
|
|
let current_dep_graph = self.data.as_ref().unwrap().current.borrow();
|
|
|
|
|
|
|
|
let nodes: IndexVec<_, _> = current_dep_graph.nodes.iter().map(|dep_node| {
|
|
|
|
let fingerprint = fingerprints.get(dep_node)
|
|
|
|
.cloned()
|
|
|
|
.unwrap_or(Fingerprint::zero());
|
|
|
|
(*dep_node, fingerprint)
|
|
|
|
}).collect();
|
|
|
|
|
|
|
|
let total_edge_count: usize = current_dep_graph.edges.iter()
|
|
|
|
.map(|v| v.len())
|
|
|
|
.sum();
|
|
|
|
|
|
|
|
let mut edge_list_indices = IndexVec::with_capacity(nodes.len());
|
|
|
|
let mut edge_list_data = Vec::with_capacity(total_edge_count);
|
|
|
|
|
|
|
|
for (current_dep_node_index, edges) in current_dep_graph.edges.iter_enumerated() {
|
|
|
|
let start = edge_list_data.len() as u32;
|
|
|
|
// This should really just be a memcpy :/
|
|
|
|
edge_list_data.extend(edges.iter().map(|i| SerializedDepNodeIndex(i.index)));
|
|
|
|
let end = edge_list_data.len() as u32;
|
|
|
|
|
|
|
|
debug_assert_eq!(current_dep_node_index.index(), edge_list_indices.len());
|
|
|
|
edge_list_indices.push((start, end));
|
|
|
|
}
|
|
|
|
|
|
|
|
debug_assert!(edge_list_data.len() <= ::std::u32::MAX as usize);
|
|
|
|
debug_assert_eq!(edge_list_data.len(), total_edge_count);
|
|
|
|
|
|
|
|
SerializedDepGraph {
|
|
|
|
nodes,
|
|
|
|
edge_list_indices,
|
|
|
|
edge_list_data,
|
|
|
|
}
|
|
|
|
}
|
2017-09-25 13:51:49 +02:00
|
|
|
|
|
|
|
pub fn node_color(&self, dep_node: &DepNode) -> Option<DepNodeColor> {
|
|
|
|
self.data.as_ref().and_then(|data| data.colors.borrow().get(dep_node).cloned())
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn try_mark_green(&self,
|
|
|
|
tcx: TyCtxt,
|
|
|
|
dep_node: &DepNode)
|
|
|
|
-> Option<DepNodeIndex> {
|
|
|
|
let data = self.data.as_ref().unwrap();
|
|
|
|
|
|
|
|
debug_assert!(!data.colors.borrow().contains_key(dep_node));
|
|
|
|
debug_assert!(!data.current.borrow().node_to_node_index.contains_key(dep_node));
|
|
|
|
|
|
|
|
if dep_node.kind.is_input() {
|
|
|
|
// We should only hit try_mark_green() for inputs that do not exist
|
|
|
|
// anymore in the current compilation session. Existing inputs are
|
|
|
|
// eagerly marked as either red/green before any queries are
|
|
|
|
// executed.
|
|
|
|
debug_assert!(dep_node.extract_def_id(tcx).is_none());
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
|
|
|
let (prev_deps, prev_dep_node_index) = match data.previous.edges_from(dep_node) {
|
|
|
|
Some(prev) => {
|
|
|
|
// This DepNode and the corresponding query invocation existed
|
|
|
|
// in the previous compilation session too, so we can try to
|
|
|
|
// mark it as green by recursively marking all of its
|
|
|
|
// dependencies green.
|
|
|
|
prev
|
|
|
|
}
|
|
|
|
None => {
|
|
|
|
// This DepNode did not exist in the previous compilation session,
|
|
|
|
// so we cannot mark it as green.
|
|
|
|
return None
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut current_deps = Vec::new();
|
|
|
|
|
2017-09-26 19:43:17 +02:00
|
|
|
for &dep_dep_node_index in prev_deps {
|
|
|
|
let dep_dep_node = &data.previous.index_to_node(dep_dep_node_index);
|
|
|
|
|
2017-09-25 13:51:49 +02:00
|
|
|
let dep_dep_node_color = data.colors.borrow().get(dep_dep_node).cloned();
|
|
|
|
match dep_dep_node_color {
|
|
|
|
Some(DepNodeColor::Green(node_index)) => {
|
|
|
|
// This dependency has been marked as green before, we are
|
|
|
|
// still fine and can continue with checking the other
|
|
|
|
// dependencies.
|
|
|
|
current_deps.push(node_index);
|
|
|
|
}
|
|
|
|
Some(DepNodeColor::Red) => {
|
|
|
|
// We found a dependency the value of which has changed
|
|
|
|
// compared to the previous compilation session. We cannot
|
|
|
|
// mark the DepNode as green and also don't need to bother
|
|
|
|
// with checking any of the other dependencies.
|
|
|
|
return None
|
|
|
|
}
|
|
|
|
None => {
|
2017-09-26 19:43:17 +02:00
|
|
|
if dep_dep_node.kind.is_input() {
|
|
|
|
// This input does not exist anymore.
|
|
|
|
debug_assert!(dep_dep_node.extract_def_id(tcx).is_none());
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
2017-09-25 13:51:49 +02:00
|
|
|
// We don't know the state of this dependency. Let's try to
|
|
|
|
// mark it green.
|
|
|
|
if let Some(node_index) = self.try_mark_green(tcx, dep_dep_node) {
|
|
|
|
current_deps.push(node_index);
|
|
|
|
} else {
|
2017-09-26 19:43:17 +02:00
|
|
|
// We failed to mark it green, so we try to force the query.
|
|
|
|
if ::ty::maps::force_from_dep_node(tcx, dep_dep_node) {
|
|
|
|
let dep_dep_node_color = data.colors.borrow().get(dep_dep_node).cloned();
|
|
|
|
match dep_dep_node_color {
|
|
|
|
Some(DepNodeColor::Green(node_index)) => {
|
|
|
|
current_deps.push(node_index);
|
|
|
|
}
|
|
|
|
Some(DepNodeColor::Red) => {
|
|
|
|
return None
|
|
|
|
}
|
|
|
|
None => {
|
|
|
|
bug!("try_mark_green() - Forcing the DepNode \
|
|
|
|
should have set its color")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// The DepNode could not be forced.
|
|
|
|
return None
|
|
|
|
}
|
2017-09-25 13:51:49 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-26 19:43:17 +02:00
|
|
|
|
2017-09-25 13:51:49 +02:00
|
|
|
// If we got here without hitting a `return` that means that all
|
|
|
|
// dependencies of this DepNode could be marked as green. Therefore we
|
|
|
|
// can also mark this DepNode as green. We do so by...
|
|
|
|
|
|
|
|
// ... allocating an entry for it in the current dependency graph and
|
|
|
|
// adding all the appropriate edges imported from the previous graph ...
|
|
|
|
let node_index_new = data.current
|
|
|
|
.borrow_mut()
|
|
|
|
.alloc_node(*dep_node,
|
|
|
|
current_deps.iter().map(|n| n.new).collect());
|
|
|
|
let dep_node_index_legacy = {
|
|
|
|
let mut legacy_graph = data.edges.borrow_mut();
|
|
|
|
legacy_graph.push_task(*dep_node);
|
|
|
|
for node_index in current_deps.into_iter().map(|n| n.legacy) {
|
|
|
|
legacy_graph.read_index(node_index);
|
|
|
|
}
|
|
|
|
legacy_graph.pop_task(*dep_node)
|
|
|
|
};
|
|
|
|
|
|
|
|
// ... copying the fingerprint from the previous graph too, so we don't
|
|
|
|
// have to recompute it ...
|
|
|
|
let fingerprint = data.previous.fingerprint_by_index(prev_dep_node_index);
|
|
|
|
assert!(self.fingerprints
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(*dep_node, fingerprint)
|
|
|
|
.is_none());
|
|
|
|
|
|
|
|
let node_index = DepNodeIndex {
|
|
|
|
legacy: dep_node_index_legacy,
|
|
|
|
new: node_index_new,
|
|
|
|
};
|
|
|
|
|
|
|
|
// ... and finally storing a "Green" entry in the color map.
|
|
|
|
assert!(data.colors
|
|
|
|
.borrow_mut()
|
|
|
|
.insert(*dep_node, DepNodeColor::Green(node_index))
|
|
|
|
.is_none());
|
|
|
|
|
|
|
|
Some(node_index)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Used in various assertions
|
|
|
|
pub fn is_green(&self, dep_node_index: DepNodeIndex) -> bool {
|
|
|
|
let dep_node = self.data.as_ref().unwrap().current.borrow().nodes[dep_node_index.new];
|
|
|
|
self.data.as_ref().unwrap().colors.borrow().get(&dep_node).map(|&color| {
|
|
|
|
match color {
|
|
|
|
DepNodeColor::Red => false,
|
|
|
|
DepNodeColor::Green(_) => true,
|
|
|
|
}
|
|
|
|
}).unwrap_or(false)
|
|
|
|
}
|
2016-07-21 12:33:23 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
/// A "work product" is an intermediate result that we save into the
|
|
|
|
/// incremental directory for later re-use. The primary example are
|
|
|
|
/// the object files that we save for each partition at code
|
|
|
|
/// generation time.
|
|
|
|
///
|
|
|
|
/// Each work product is associated with a dep-node, representing the
|
|
|
|
/// process that produced the work-product. If that dep-node is found
|
|
|
|
/// to be dirty when we load up, then we will delete the work-product
|
2016-07-22 10:39:30 -04:00
|
|
|
/// at load time. If the work-product is found to be clean, then we
|
2016-07-21 12:33:23 -04:00
|
|
|
/// will keep a record in the `previous_work_products` list.
|
|
|
|
///
|
|
|
|
/// In addition, work products have an associated hash. This hash is
|
|
|
|
/// an extra hash that can be used to decide if the work-product from
|
|
|
|
/// a previous compilation can be re-used (in addition to the dirty
|
|
|
|
/// edges check).
|
|
|
|
///
|
|
|
|
/// As the primary example, consider the object files we generate for
|
|
|
|
/// each partition. In the first run, we create partitions based on
|
|
|
|
/// the symbols that need to be compiled. For each partition P, we
|
|
|
|
/// hash the symbols in P and create a `WorkProduct` record associated
|
|
|
|
/// with `DepNode::TransPartition(P)`; the hash is the set of symbols
|
|
|
|
/// in P.
|
|
|
|
///
|
|
|
|
/// The next time we compile, if the `DepNode::TransPartition(P)` is
|
|
|
|
/// judged to be clean (which means none of the things we read to
|
|
|
|
/// generate the partition were found to be dirty), it will be loaded
|
|
|
|
/// into previous work products. We will then regenerate the set of
|
|
|
|
/// symbols in the partition P and hash them (note that new symbols
|
|
|
|
/// may be added -- for example, new monomorphizations -- even if
|
|
|
|
/// nothing in P changed!). We will compare that hash against the
|
|
|
|
/// previous hash. If it matches up, we can reuse the object file.
|
|
|
|
#[derive(Clone, Debug, RustcEncodable, RustcDecodable)]
|
|
|
|
pub struct WorkProduct {
|
2017-06-23 16:37:12 +02:00
|
|
|
pub cgu_name: String,
|
2016-07-25 10:51:14 -04:00
|
|
|
/// Extra hash used to decide if work-product is still suitable;
|
2016-07-21 12:33:23 -04:00
|
|
|
/// note that this is *not* a hash of the work-product itself.
|
|
|
|
/// See documentation on `WorkProduct` type for an example.
|
|
|
|
pub input_hash: u64,
|
|
|
|
|
2016-07-25 10:51:14 -04:00
|
|
|
/// Saved files associated with this CGU
|
|
|
|
pub saved_files: Vec<(OutputType, String)>,
|
2016-03-28 17:37:34 -04:00
|
|
|
}
|
2017-08-21 16:44:05 +02:00
|
|
|
|
|
|
|
pub(super) struct CurrentDepGraph {
|
|
|
|
nodes: IndexVec<DepNodeIndexNew, DepNode>,
|
|
|
|
edges: IndexVec<DepNodeIndexNew, Vec<DepNodeIndexNew>>,
|
|
|
|
node_to_node_index: FxHashMap<DepNode, DepNodeIndexNew>,
|
|
|
|
|
|
|
|
task_stack: Vec<OpenTask>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl CurrentDepGraph {
|
|
|
|
fn new() -> CurrentDepGraph {
|
|
|
|
CurrentDepGraph {
|
|
|
|
nodes: IndexVec::new(),
|
|
|
|
edges: IndexVec::new(),
|
|
|
|
node_to_node_index: FxHashMap(),
|
|
|
|
task_stack: Vec::new(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(super) fn push_ignore(&mut self) {
|
|
|
|
self.task_stack.push(OpenTask::Ignore);
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(super) fn pop_ignore(&mut self) {
|
|
|
|
let popped_node = self.task_stack.pop().unwrap();
|
|
|
|
debug_assert_eq!(popped_node, OpenTask::Ignore);
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(super) fn push_task(&mut self, key: DepNode) {
|
|
|
|
self.task_stack.push(OpenTask::Regular {
|
|
|
|
node: key,
|
|
|
|
reads: Vec::new(),
|
|
|
|
read_set: FxHashSet(),
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(super) fn pop_task(&mut self, key: DepNode) -> DepNodeIndexNew {
|
|
|
|
let popped_node = self.task_stack.pop().unwrap();
|
|
|
|
|
|
|
|
if let OpenTask::Regular {
|
|
|
|
node,
|
|
|
|
read_set: _,
|
|
|
|
reads
|
|
|
|
} = popped_node {
|
|
|
|
debug_assert_eq!(node, key);
|
|
|
|
self.alloc_node(node, reads)
|
|
|
|
} else {
|
|
|
|
bug!("pop_task() - Expected regular task to be popped")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn push_anon_task(&mut self) {
|
|
|
|
self.task_stack.push(OpenTask::Anon {
|
|
|
|
reads: Vec::new(),
|
|
|
|
read_set: FxHashSet(),
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-09-25 12:25:41 +02:00
|
|
|
fn pop_anon_task(&mut self, kind: DepKind) -> (Option<DepNode>, DepNodeIndexNew) {
|
2017-08-21 16:44:05 +02:00
|
|
|
let popped_node = self.task_stack.pop().unwrap();
|
|
|
|
|
|
|
|
if let OpenTask::Anon {
|
|
|
|
read_set: _,
|
|
|
|
reads
|
|
|
|
} = popped_node {
|
|
|
|
let mut fingerprint = Fingerprint::zero();
|
|
|
|
let mut hasher = StableHasher::new();
|
|
|
|
|
|
|
|
for &read in reads.iter() {
|
|
|
|
let read_dep_node = self.nodes[read];
|
|
|
|
|
|
|
|
::std::mem::discriminant(&read_dep_node.kind).hash(&mut hasher);
|
|
|
|
|
|
|
|
// Fingerprint::combine() is faster than sending Fingerprint
|
|
|
|
// through the StableHasher (at least as long as StableHasher
|
|
|
|
// is so slow).
|
|
|
|
fingerprint = fingerprint.combine(read_dep_node.hash);
|
|
|
|
}
|
|
|
|
|
|
|
|
fingerprint = fingerprint.combine(hasher.finish());
|
|
|
|
|
|
|
|
let target_dep_node = DepNode {
|
|
|
|
kind,
|
|
|
|
hash: fingerprint,
|
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(&index) = self.node_to_node_index.get(&target_dep_node) {
|
2017-09-25 12:25:41 +02:00
|
|
|
(None, index)
|
|
|
|
} else {
|
|
|
|
(Some(target_dep_node), self.alloc_node(target_dep_node, reads))
|
2017-08-21 16:44:05 +02:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
bug!("pop_anon_task() - Expected anonymous task to be popped")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn read_index(&mut self, source: DepNodeIndexNew) {
|
|
|
|
match self.task_stack.last_mut() {
|
|
|
|
Some(&mut OpenTask::Regular {
|
|
|
|
ref mut reads,
|
|
|
|
ref mut read_set,
|
|
|
|
node: _,
|
|
|
|
}) => {
|
|
|
|
if read_set.insert(source) {
|
|
|
|
reads.push(source);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Some(&mut OpenTask::Anon {
|
|
|
|
ref mut reads,
|
|
|
|
ref mut read_set,
|
|
|
|
}) => {
|
|
|
|
if read_set.insert(source) {
|
|
|
|
reads.push(source);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Some(&mut OpenTask::Ignore) | None => {
|
|
|
|
// ignore
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn alloc_node(&mut self,
|
|
|
|
dep_node: DepNode,
|
|
|
|
edges: Vec<DepNodeIndexNew>)
|
|
|
|
-> DepNodeIndexNew {
|
|
|
|
debug_assert_eq!(self.edges.len(), self.nodes.len());
|
|
|
|
debug_assert_eq!(self.node_to_node_index.len(), self.nodes.len());
|
|
|
|
debug_assert!(!self.node_to_node_index.contains_key(&dep_node));
|
|
|
|
let dep_node_index = DepNodeIndexNew::new(self.nodes.len());
|
|
|
|
self.nodes.push(dep_node);
|
|
|
|
self.node_to_node_index.insert(dep_node, dep_node_index);
|
|
|
|
self.edges.push(edges);
|
|
|
|
dep_node_index
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
|
|
|
|
pub(super) struct DepNodeIndexNew {
|
|
|
|
index: u32,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Idx for DepNodeIndexNew {
|
2017-09-27 16:31:31 -03:00
|
|
|
fn new(v: usize) -> DepNodeIndexNew {
|
|
|
|
assert!((v & 0xFFFF_FFFF) == v);
|
|
|
|
DepNodeIndexNew { index: v as u32 }
|
2017-08-21 16:44:05 +02:00
|
|
|
}
|
2017-09-27 16:31:31 -03:00
|
|
|
|
2017-08-21 16:44:05 +02:00
|
|
|
fn index(self) -> usize {
|
2017-09-27 16:31:31 -03:00
|
|
|
self.index as usize
|
2017-08-21 16:44:05 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl DepNodeIndexNew {
|
|
|
|
const INVALID: DepNodeIndexNew = DepNodeIndexNew {
|
|
|
|
index: ::std::u32::MAX,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
|
|
enum OpenTask {
|
|
|
|
Regular {
|
|
|
|
node: DepNode,
|
|
|
|
reads: Vec<DepNodeIndexNew>,
|
|
|
|
read_set: FxHashSet<DepNodeIndexNew>,
|
|
|
|
},
|
|
|
|
Anon {
|
|
|
|
reads: Vec<DepNodeIndexNew>,
|
|
|
|
read_set: FxHashSet<DepNodeIndexNew>,
|
|
|
|
},
|
|
|
|
Ignore,
|
|
|
|
}
|