1
Fork 0

get rid of RefCell in TransitiveRelation

This commit is contained in:
SparrowLii 2022-07-25 11:06:22 +08:00
parent a9bb589cd6
commit 5d9e4d07fc
9 changed files with 179 additions and 118 deletions

View file

@ -1,5 +1,5 @@
use rustc_data_structures::frozen::Frozen; use rustc_data_structures::frozen::Frozen;
use rustc_data_structures::transitive_relation::TransitiveRelation; use rustc_data_structures::transitive_relation::{TransitiveRelation, TransitiveRelationBuilder};
use rustc_infer::infer::canonical::QueryRegionConstraints; use rustc_infer::infer::canonical::QueryRegionConstraints;
use rustc_infer::infer::outlives; use rustc_infer::infer::outlives;
use rustc_infer::infer::outlives::env::RegionBoundPairs; use rustc_infer::infer::outlives::env::RegionBoundPairs;
@ -61,25 +61,13 @@ pub(crate) fn create<'tcx>(
constraints, constraints,
universal_regions: universal_regions.clone(), universal_regions: universal_regions.clone(),
region_bound_pairs: Default::default(), region_bound_pairs: Default::default(),
relations: UniversalRegionRelations {
universal_regions: universal_regions.clone(),
outlives: Default::default(), outlives: Default::default(),
inverse_outlives: Default::default(), inverse_outlives: Default::default(),
},
} }
.create() .create()
} }
impl UniversalRegionRelations<'_> { impl UniversalRegionRelations<'_> {
/// Records in the `outlives_relation` (and
/// `inverse_outlives_relation`) that `fr_a: fr_b`. Invoked by the
/// builder below.
fn relate_universal_regions(&mut self, fr_a: RegionVid, fr_b: RegionVid) {
debug!("relate_universal_regions: fr_a={:?} outlives fr_b={:?}", fr_a, fr_b);
self.outlives.add(fr_a, fr_b);
self.inverse_outlives.add(fr_b, fr_a);
}
/// Given two universal regions, returns the postdominating /// Given two universal regions, returns the postdominating
/// upper-bound (effectively the least upper bound). /// upper-bound (effectively the least upper bound).
/// ///
@ -216,11 +204,20 @@ struct UniversalRegionRelationsBuilder<'this, 'tcx> {
constraints: &'this mut MirTypeckRegionConstraints<'tcx>, constraints: &'this mut MirTypeckRegionConstraints<'tcx>,
// outputs: // outputs:
relations: UniversalRegionRelations<'tcx>, outlives: TransitiveRelationBuilder<RegionVid>,
inverse_outlives: TransitiveRelationBuilder<RegionVid>,
region_bound_pairs: RegionBoundPairs<'tcx>, region_bound_pairs: RegionBoundPairs<'tcx>,
} }
impl<'tcx> UniversalRegionRelationsBuilder<'_, 'tcx> { impl<'tcx> UniversalRegionRelationsBuilder<'_, 'tcx> {
/// Records in the `outlives_relation` (and
/// `inverse_outlives_relation`) that `fr_a: fr_b`.
fn relate_universal_regions(&mut self, fr_a: RegionVid, fr_b: RegionVid) {
debug!("relate_universal_regions: fr_a={:?} outlives fr_b={:?}", fr_a, fr_b);
self.outlives.add(fr_a, fr_b);
self.inverse_outlives.add(fr_b, fr_a);
}
pub(crate) fn create(mut self) -> CreateResult<'tcx> { pub(crate) fn create(mut self) -> CreateResult<'tcx> {
let unnormalized_input_output_tys = self let unnormalized_input_output_tys = self
.universal_regions .universal_regions
@ -292,9 +289,9 @@ impl<'tcx> UniversalRegionRelationsBuilder<'_, 'tcx> {
let fr_fn_body = self.universal_regions.fr_fn_body; let fr_fn_body = self.universal_regions.fr_fn_body;
for fr in self.universal_regions.universal_regions() { for fr in self.universal_regions.universal_regions() {
debug!("build: relating free region {:?} to itself and to 'static", fr); debug!("build: relating free region {:?} to itself and to 'static", fr);
self.relations.relate_universal_regions(fr, fr); self.relate_universal_regions(fr, fr);
self.relations.relate_universal_regions(fr_static, fr); self.relate_universal_regions(fr_static, fr);
self.relations.relate_universal_regions(fr, fr_fn_body); self.relate_universal_regions(fr, fr_fn_body);
} }
for data in &constraint_sets { for data in &constraint_sets {
@ -313,7 +310,11 @@ impl<'tcx> UniversalRegionRelationsBuilder<'_, 'tcx> {
} }
CreateResult { CreateResult {
universal_region_relations: Frozen::freeze(self.relations), universal_region_relations: Frozen::freeze(UniversalRegionRelations {
universal_regions: self.universal_regions,
outlives: self.outlives.freeze(),
inverse_outlives: self.inverse_outlives.freeze(),
}),
region_bound_pairs: self.region_bound_pairs, region_bound_pairs: self.region_bound_pairs,
normalized_inputs_and_output, normalized_inputs_and_output,
} }
@ -356,7 +357,7 @@ impl<'tcx> UniversalRegionRelationsBuilder<'_, 'tcx> {
// The bound says that `r1 <= r2`; we store `r2: r1`. // The bound says that `r1 <= r2`; we store `r2: r1`.
let r1 = self.universal_regions.to_region_vid(r1); let r1 = self.universal_regions.to_region_vid(r1);
let r2 = self.universal_regions.to_region_vid(r2); let r2 = self.universal_regions.to_region_vid(r2);
self.relations.relate_universal_regions(r2, r1); self.relate_universal_regions(r2, r1);
} }
OutlivesBound::RegionSubParam(r_a, param_b) => { OutlivesBound::RegionSubParam(r_a, param_b) => {

View file

@ -1,42 +1,54 @@
use crate::frozen::Frozen;
use crate::fx::FxIndexSet; use crate::fx::FxIndexSet;
use crate::sync::Lock;
use rustc_index::bit_set::BitMatrix; use rustc_index::bit_set::BitMatrix;
use std::fmt::Debug; use std::fmt::Debug;
use std::hash::Hash; use std::hash::Hash;
use std::mem; use std::mem;
use std::ops::Deref;
#[cfg(test)] #[cfg(test)]
mod tests; mod tests;
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub struct TransitiveRelation<T> { pub struct TransitiveRelationBuilder<T> {
// List of elements. This is used to map from a T to a usize. // List of elements. This is used to map from a T to a usize.
elements: FxIndexSet<T>, elements: FxIndexSet<T>,
// List of base edges in the graph. Require to compute transitive // List of base edges in the graph. Require to compute transitive
// closure. // closure.
edges: Vec<Edge>, edges: Vec<Edge>,
}
// This is a cached transitive closure derived from the edges. #[derive(Debug)]
// Currently, we build it lazily and just throw out any existing pub struct TransitiveRelation<T> {
// copy whenever a new edge is added. (The Lock is to permit // Frozen transitive relation elements and edges.
// the lazy computation.) This is kind of silly, except for the builder: Frozen<TransitiveRelationBuilder<T>>,
// fact its size is tied to `self.elements.len()`, so I wanted to
// wait before building it up to avoid reallocating as new edges // Cached transitive closure derived from the edges.
// are added with new elements. Perhaps better would be to ask the closure: Frozen<BitMatrix<usize, usize>>,
// user for a batch of edges to minimize this effect, but I }
// already wrote the code this way. :P -nmatsakis
closure: Lock<Option<BitMatrix<usize, usize>>>, impl<T> Deref for TransitiveRelation<T> {
type Target = Frozen<TransitiveRelationBuilder<T>>;
fn deref(&self) -> &Self::Target {
&self.builder
}
}
impl<T: Clone> Clone for TransitiveRelation<T> {
fn clone(&self) -> Self {
TransitiveRelation {
builder: Frozen::freeze(self.builder.deref().clone()),
closure: Frozen::freeze(self.closure.deref().clone()),
}
}
} }
// HACK(eddyb) manual impl avoids `Default` bound on `T`. // HACK(eddyb) manual impl avoids `Default` bound on `T`.
impl<T: Eq + Hash> Default for TransitiveRelation<T> { impl<T: Eq + Hash> Default for TransitiveRelationBuilder<T> {
fn default() -> Self { fn default() -> Self {
TransitiveRelation { TransitiveRelationBuilder { elements: Default::default(), edges: Default::default() }
elements: Default::default(),
edges: Default::default(),
closure: Default::default(),
}
} }
} }
@ -49,7 +61,7 @@ struct Edge {
target: Index, target: Index,
} }
impl<T: Eq + Hash + Copy> TransitiveRelation<T> { impl<T: Eq + Hash + Copy> TransitiveRelationBuilder<T> {
pub fn is_empty(&self) -> bool { pub fn is_empty(&self) -> bool {
self.edges.is_empty() self.edges.is_empty()
} }
@ -63,23 +75,19 @@ impl<T: Eq + Hash + Copy> TransitiveRelation<T> {
} }
fn add_index(&mut self, a: T) -> Index { fn add_index(&mut self, a: T) -> Index {
let (index, added) = self.elements.insert_full(a); let (index, _added) = self.elements.insert_full(a);
if added {
// if we changed the dimensions, clear the cache
*self.closure.get_mut() = None;
}
Index(index) Index(index)
} }
/// Applies the (partial) function to each edge and returns a new /// Applies the (partial) function to each edge and returns a new
/// relation. If `f` returns `None` for any end-point, returns /// relation builder. If `f` returns `None` for any end-point,
/// `None`. /// returns `None`.
pub fn maybe_map<F, U>(&self, mut f: F) -> Option<TransitiveRelation<U>> pub fn maybe_map<F, U>(&self, mut f: F) -> Option<TransitiveRelationBuilder<U>>
where where
F: FnMut(T) -> Option<U>, F: FnMut(T) -> Option<U>,
U: Clone + Debug + Eq + Hash + Copy, U: Clone + Debug + Eq + Hash + Copy,
{ {
let mut result = TransitiveRelation::default(); let mut result = TransitiveRelationBuilder::default();
for edge in &self.edges { for edge in &self.edges {
result.add(f(self.elements[edge.source.0])?, f(self.elements[edge.target.0])?); result.add(f(self.elements[edge.source.0])?, f(self.elements[edge.target.0])?);
} }
@ -93,12 +101,40 @@ impl<T: Eq + Hash + Copy> TransitiveRelation<T> {
let edge = Edge { source: a, target: b }; let edge = Edge { source: a, target: b };
if !self.edges.contains(&edge) { if !self.edges.contains(&edge) {
self.edges.push(edge); self.edges.push(edge);
// added an edge, clear the cache
*self.closure.get_mut() = None;
} }
} }
/// Compute the transitive closure derived from the edges, and converted to
/// the final result. After this, all elements will be immutable to maintain
/// the correctness of the result.
pub fn freeze(self) -> TransitiveRelation<T> {
let mut matrix = BitMatrix::new(self.elements.len(), self.elements.len());
let mut changed = true;
while changed {
changed = false;
for edge in &self.edges {
// add an edge from S -> T
changed |= matrix.insert(edge.source.0, edge.target.0);
// add all outgoing edges from T into S
changed |= matrix.union_rows(edge.target.0, edge.source.0);
}
}
TransitiveRelation { builder: Frozen::freeze(self), closure: Frozen::freeze(matrix) }
}
}
impl<T: Eq + Hash + Copy> TransitiveRelation<T> {
/// Applies the (partial) function to each edge and returns a new
/// relation including transitive closures.
pub fn maybe_map<F, U>(&self, f: F) -> Option<TransitiveRelation<U>>
where
F: FnMut(T) -> Option<U>,
U: Clone + Debug + Eq + Hash + Copy,
{
Some(self.builder.maybe_map(f)?.freeze())
}
/// Checks whether `a < target` (transitively) /// Checks whether `a < target` (transitively)
pub fn contains(&self, a: T, b: T) -> bool { pub fn contains(&self, a: T, b: T) -> bool {
match (self.index(a), self.index(b)) { match (self.index(a), self.index(b)) {
@ -322,30 +358,7 @@ impl<T: Eq + Hash + Copy> TransitiveRelation<T> {
where where
OP: FnOnce(&BitMatrix<usize, usize>) -> R, OP: FnOnce(&BitMatrix<usize, usize>) -> R,
{ {
let mut closure_cell = self.closure.borrow_mut(); op(&self.closure)
let mut closure = closure_cell.take();
if closure.is_none() {
closure = Some(self.compute_closure());
}
let result = op(closure.as_ref().unwrap());
*closure_cell = closure;
result
}
fn compute_closure(&self) -> BitMatrix<usize, usize> {
let mut matrix = BitMatrix::new(self.elements.len(), self.elements.len());
let mut changed = true;
while changed {
changed = false;
for edge in &self.edges {
// add an edge from S -> T
changed |= matrix.insert(edge.source.0, edge.target.0);
// add all outgoing edges from T into S
changed |= matrix.union_rows(edge.target.0, edge.source.0);
}
}
matrix
} }
/// Lists all the base edges in the graph: the initial _non-transitive_ set of element /// Lists all the base edges in the graph: the initial _non-transitive_ set of element

View file

@ -10,9 +10,10 @@ impl<T: Eq + Hash + Copy> TransitiveRelation<T> {
#[test] #[test]
fn test_one_step() { fn test_one_step() {
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "b"); relation.add("a", "b");
relation.add("a", "c"); relation.add("a", "c");
let relation = relation.freeze();
assert!(relation.contains("a", "c")); assert!(relation.contains("a", "c"));
assert!(relation.contains("a", "b")); assert!(relation.contains("a", "b"));
assert!(!relation.contains("b", "a")); assert!(!relation.contains("b", "a"));
@ -21,7 +22,7 @@ fn test_one_step() {
#[test] #[test]
fn test_many_steps() { fn test_many_steps() {
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "b"); relation.add("a", "b");
relation.add("a", "c"); relation.add("a", "c");
relation.add("a", "f"); relation.add("a", "f");
@ -31,6 +32,7 @@ fn test_many_steps() {
relation.add("b", "e"); relation.add("b", "e");
relation.add("e", "g"); relation.add("e", "g");
let relation = relation.freeze();
assert!(relation.contains("a", "b")); assert!(relation.contains("a", "b"));
assert!(relation.contains("a", "c")); assert!(relation.contains("a", "c"));
@ -51,9 +53,10 @@ fn mubs_triangle() {
// ^ // ^
// | // |
// b // b
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "tcx"); relation.add("a", "tcx");
relation.add("b", "tcx"); relation.add("b", "tcx");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["tcx"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["tcx"]);
assert_eq!(relation.parents("a"), vec!["tcx"]); assert_eq!(relation.parents("a"), vec!["tcx"]);
assert_eq!(relation.parents("b"), vec!["tcx"]); assert_eq!(relation.parents("b"), vec!["tcx"]);
@ -72,7 +75,7 @@ fn mubs_best_choice1() {
// need the second pare down call to get the right result (after // need the second pare down call to get the right result (after
// intersection, we have [1, 2], but 2 -> 1). // intersection, we have [1, 2], but 2 -> 1).
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("0", "1"); relation.add("0", "1");
relation.add("0", "2"); relation.add("0", "2");
@ -80,6 +83,7 @@ fn mubs_best_choice1() {
relation.add("3", "1"); relation.add("3", "1");
relation.add("3", "2"); relation.add("3", "2");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["2"]); assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["2"]);
assert_eq!(relation.parents("0"), vec!["2"]); assert_eq!(relation.parents("0"), vec!["2"]);
@ -99,7 +103,7 @@ fn mubs_best_choice2() {
// Like the preceding test, but in this case intersection is [2, // Like the preceding test, but in this case intersection is [2,
// 1], and hence we rely on the first pare down call. // 1], and hence we rely on the first pare down call.
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("0", "1"); relation.add("0", "1");
relation.add("0", "2"); relation.add("0", "2");
@ -107,6 +111,7 @@ fn mubs_best_choice2() {
relation.add("3", "1"); relation.add("3", "1");
relation.add("3", "2"); relation.add("3", "2");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1"]); assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1"]);
assert_eq!(relation.parents("0"), vec!["1"]); assert_eq!(relation.parents("0"), vec!["1"]);
@ -118,12 +123,13 @@ fn mubs_best_choice2() {
fn mubs_no_best_choice() { fn mubs_no_best_choice() {
// in this case, the intersection yields [1, 2], and the "pare // in this case, the intersection yields [1, 2], and the "pare
// down" calls find nothing to remove. // down" calls find nothing to remove.
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("0", "1"); relation.add("0", "1");
relation.add("0", "2"); relation.add("0", "2");
relation.add("3", "1"); relation.add("3", "1");
relation.add("3", "2"); relation.add("3", "2");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1", "2"]); assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1", "2"]);
assert_eq!(relation.parents("0"), vec!["1", "2"]); assert_eq!(relation.parents("0"), vec!["1", "2"]);
@ -135,7 +141,7 @@ fn mubs_best_choice_scc() {
// in this case, 1 and 2 form a cycle; we pick arbitrarily (but // in this case, 1 and 2 form a cycle; we pick arbitrarily (but
// consistently). // consistently).
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("0", "1"); relation.add("0", "1");
relation.add("0", "2"); relation.add("0", "2");
@ -144,6 +150,7 @@ fn mubs_best_choice_scc() {
relation.add("3", "1"); relation.add("3", "1");
relation.add("3", "2"); relation.add("3", "2");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1"]); assert_eq!(relation.minimal_upper_bounds("0", "3"), vec!["1"]);
assert_eq!(relation.parents("0"), vec!["1"]); assert_eq!(relation.parents("0"), vec!["1"]);
@ -157,13 +164,14 @@ fn pdub_crisscross() {
// /\ | // /\ |
// b -> b1 ---+ // b -> b1 ---+
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "a1"); relation.add("a", "a1");
relation.add("a", "b1"); relation.add("a", "b1");
relation.add("b", "a1"); relation.add("b", "a1");
relation.add("b", "b1"); relation.add("b", "b1");
relation.add("a1", "x"); relation.add("a1", "x");
relation.add("b1", "x"); relation.add("b1", "x");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["a1", "b1"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["a1", "b1"]);
assert_eq!(relation.postdom_upper_bound("a", "b"), Some("x")); assert_eq!(relation.postdom_upper_bound("a", "b"), Some("x"));
@ -179,7 +187,7 @@ fn pdub_crisscross_more() {
// /\ /\ | // /\ /\ |
// b -> b1 -> b2 ---------+ // b -> b1 -> b2 ---------+
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "a1"); relation.add("a", "a1");
relation.add("a", "b1"); relation.add("a", "b1");
relation.add("b", "a1"); relation.add("b", "a1");
@ -194,6 +202,7 @@ fn pdub_crisscross_more() {
relation.add("a3", "x"); relation.add("a3", "x");
relation.add("b2", "x"); relation.add("b2", "x");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["a1", "b1"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["a1", "b1"]);
assert_eq!(relation.minimal_upper_bounds("a1", "b1"), vec!["a2", "b2"]); assert_eq!(relation.minimal_upper_bounds("a1", "b1"), vec!["a2", "b2"]);
@ -210,11 +219,12 @@ fn pdub_lub() {
// | // |
// b -> b1 ---+ // b -> b1 ---+
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "a1"); relation.add("a", "a1");
relation.add("b", "b1"); relation.add("b", "b1");
relation.add("a1", "x"); relation.add("a1", "x");
relation.add("b1", "x"); relation.add("b1", "x");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["x"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["x"]);
assert_eq!(relation.postdom_upper_bound("a", "b"), Some("x")); assert_eq!(relation.postdom_upper_bound("a", "b"), Some("x"));
@ -233,10 +243,11 @@ fn mubs_intermediate_node_on_one_side_only() {
// b // b
// "digraph { a -> c -> d; b -> d; }", // "digraph { a -> c -> d; b -> d; }",
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "c"); relation.add("a", "c");
relation.add("c", "d"); relation.add("c", "d");
relation.add("b", "d"); relation.add("b", "d");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["d"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["d"]);
} }
@ -252,12 +263,13 @@ fn mubs_scc_1() {
// b // b
// "digraph { a -> c -> d; d -> c; a -> d; b -> d; }", // "digraph { a -> c -> d; d -> c; a -> d; b -> d; }",
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "c"); relation.add("a", "c");
relation.add("c", "d"); relation.add("c", "d");
relation.add("d", "c"); relation.add("d", "c");
relation.add("a", "d"); relation.add("a", "d");
relation.add("b", "d"); relation.add("b", "d");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]);
} }
@ -272,12 +284,13 @@ fn mubs_scc_2() {
// +--- b // +--- b
// "digraph { a -> c -> d; d -> c; b -> d; b -> c; }", // "digraph { a -> c -> d; d -> c; b -> d; b -> c; }",
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "c"); relation.add("a", "c");
relation.add("c", "d"); relation.add("c", "d");
relation.add("d", "c"); relation.add("d", "c");
relation.add("b", "d"); relation.add("b", "d");
relation.add("b", "c"); relation.add("b", "c");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]);
} }
@ -292,13 +305,14 @@ fn mubs_scc_3() {
// b ---+ // b ---+
// "digraph { a -> c -> d -> e -> c; b -> d; b -> e; }", // "digraph { a -> c -> d -> e -> c; b -> d; b -> e; }",
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "c"); relation.add("a", "c");
relation.add("c", "d"); relation.add("c", "d");
relation.add("d", "e"); relation.add("d", "e");
relation.add("e", "c"); relation.add("e", "c");
relation.add("b", "d"); relation.add("b", "d");
relation.add("b", "e"); relation.add("b", "e");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]);
} }
@ -314,13 +328,14 @@ fn mubs_scc_4() {
// b ---+ // b ---+
// "digraph { a -> c -> d -> e -> c; a -> d; b -> e; }" // "digraph { a -> c -> d -> e -> c; a -> d; b -> e; }"
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
relation.add("a", "c"); relation.add("a", "c");
relation.add("c", "d"); relation.add("c", "d");
relation.add("d", "e"); relation.add("d", "e");
relation.add("e", "c"); relation.add("e", "c");
relation.add("a", "d"); relation.add("a", "d");
relation.add("b", "e"); relation.add("b", "e");
let relation = relation.freeze();
assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]); assert_eq!(relation.minimal_upper_bounds("a", "b"), vec!["c"]);
} }
@ -352,10 +367,11 @@ fn parent() {
(1, /*->*/ 3), (1, /*->*/ 3),
]; ];
let mut relation = TransitiveRelation::default(); let mut relation = TransitiveRelationBuilder::default();
for (a, b) in pairs { for (a, b) in pairs {
relation.add(a, b); relation.add(a, b);
} }
let relation = relation.freeze();
let p = relation.postdom_parent(3); let p = relation.postdom_parent(3);
assert_eq!(p, Some(0)); assert_eq!(p, Some(0));

View file

@ -27,13 +27,13 @@ impl<'a, 'tcx> RegionRelations<'a, 'tcx> {
} }
} }
#[derive(Clone, Debug, Default)] #[derive(Clone, Debug)]
pub struct FreeRegionMap<'tcx> { pub struct FreeRegionMap<'tcx> {
// Stores the relation `a < b`, where `a` and `b` are regions. // Stores the relation `a < b`, where `a` and `b` are regions.
// //
// Invariant: only free regions like `'x` or `'static` are stored // Invariant: only free regions like `'x` or `'static` are stored
// in this relation, not scopes. // in this relation, not scopes.
relation: TransitiveRelation<Region<'tcx>>, pub(crate) relation: TransitiveRelation<Region<'tcx>>,
} }
impl<'tcx> FreeRegionMap<'tcx> { impl<'tcx> FreeRegionMap<'tcx> {
@ -45,15 +45,6 @@ impl<'tcx> FreeRegionMap<'tcx> {
self.relation.is_empty() self.relation.is_empty()
} }
// Record that `'sup:'sub`. Or, put another way, `'sub <= 'sup`.
// (with the exception that `'static: 'x` is not notable)
pub fn relate_regions(&mut self, sub: Region<'tcx>, sup: Region<'tcx>) {
debug!("relate_regions(sub={:?}, sup={:?})", sub, sup);
if sub.is_free_or_static() && sup.is_free() {
self.relation.add(sub, sup)
}
}
/// Tests whether `r_a <= r_b`. /// Tests whether `r_a <= r_b`.
/// ///
/// Both regions must meet `is_free_or_static`. /// Both regions must meet `is_free_or_static`.

View file

@ -2,6 +2,7 @@ use crate::infer::free_regions::FreeRegionMap;
use crate::infer::{GenericKind, InferCtxt}; use crate::infer::{GenericKind, InferCtxt};
use crate::traits::query::OutlivesBound; use crate::traits::query::OutlivesBound;
use rustc_data_structures::fx::FxIndexSet; use rustc_data_structures::fx::FxIndexSet;
use rustc_data_structures::transitive_relation::TransitiveRelationBuilder;
use rustc_middle::ty::{self, ReEarlyBound, ReFree, ReVar, Region}; use rustc_middle::ty::{self, ReEarlyBound, ReFree, ReVar, Region};
use super::explicit_outlives_bounds; use super::explicit_outlives_bounds;
@ -51,23 +52,39 @@ pub struct OutlivesEnvironment<'tcx> {
region_bound_pairs: RegionBoundPairs<'tcx>, region_bound_pairs: RegionBoundPairs<'tcx>,
} }
/// Builder of OutlivesEnvironment. Use this structure if you need to add more outlives
/// bounds than `explicit_outlives_bounds(param_env)`.
pub struct OutlivesEnvironmentBuilder<'tcx> {
pub param_env: ty::ParamEnv<'tcx>,
region_relation: TransitiveRelationBuilder<Region<'tcx>>,
region_bound_pairs: RegionBoundPairs<'tcx>,
}
/// "Region-bound pairs" tracks outlives relations that are known to /// "Region-bound pairs" tracks outlives relations that are known to
/// be true, either because of explicit where-clauses like `T: 'a` or /// be true, either because of explicit where-clauses like `T: 'a` or
/// because of implied bounds. /// because of implied bounds.
pub type RegionBoundPairs<'tcx> = pub type RegionBoundPairs<'tcx> =
FxIndexSet<ty::OutlivesPredicate<GenericKind<'tcx>, Region<'tcx>>>; FxIndexSet<ty::OutlivesPredicate<GenericKind<'tcx>, Region<'tcx>>>;
impl<'a, 'tcx> OutlivesEnvironment<'tcx> { impl<'tcx> OutlivesEnvironment<'tcx> {
pub fn new(param_env: ty::ParamEnv<'tcx>) -> Self { /// Create a builder using `ParamEnv` and add explicit outlives bounds into it.
let mut env = OutlivesEnvironment { pub fn builder(param_env: ty::ParamEnv<'tcx>) -> OutlivesEnvironmentBuilder<'tcx> {
let mut builder = OutlivesEnvironmentBuilder {
param_env, param_env,
free_region_map: Default::default(), region_relation: Default::default(),
region_bound_pairs: Default::default(), region_bound_pairs: Default::default(),
}; };
env.add_outlives_bounds(None, explicit_outlives_bounds(param_env)); builder.add_outlives_bounds(None, explicit_outlives_bounds(param_env));
env builder
}
#[inline]
/// Create a new `OutlivesEnvironment` without extra outlives bounds.
pub fn new(param_env: ty::ParamEnv<'tcx>) -> Self {
Self::builder(param_env).build()
} }
/// Borrows current value of the `free_region_map`. /// Borrows current value of the `free_region_map`.
@ -79,6 +96,26 @@ impl<'a, 'tcx> OutlivesEnvironment<'tcx> {
pub fn region_bound_pairs(&self) -> &RegionBoundPairs<'tcx> { pub fn region_bound_pairs(&self) -> &RegionBoundPairs<'tcx> {
&self.region_bound_pairs &self.region_bound_pairs
} }
}
impl<'a, 'tcx> OutlivesEnvironmentBuilder<'tcx> {
#[inline]
pub fn build(self) -> OutlivesEnvironment<'tcx> {
OutlivesEnvironment {
param_env: self.param_env,
free_region_map: FreeRegionMap { relation: self.region_relation.freeze() },
region_bound_pairs: self.region_bound_pairs,
}
}
// Record that `'sup:'sub`. Or, put another way, `'sub <= 'sup`.
// (with the exception that `'static: 'x` is not notable)
fn relate_regions(&mut self, sub: Region<'tcx>, sup: Region<'tcx>) {
debug!("relate_regions(sub={:?}, sup={:?})", sub, sup);
if sub.is_free_or_static() && sup.is_free() {
self.region_relation.add(sub, sup)
}
}
/// Processes outlives bounds that are known to hold, whether from implied or other sources. /// Processes outlives bounds that are known to hold, whether from implied or other sources.
/// ///
@ -122,7 +159,7 @@ impl<'a, 'tcx> OutlivesEnvironment<'tcx> {
// system to be more general and to make use // system to be more general and to make use
// of *every* relationship that arises here, // of *every* relationship that arises here,
// but presently we do not.) // but presently we do not.)
self.free_region_map.relate_regions(r_a, r_b); self.relate_regions(r_a, r_b);
} }
} }
} }

View file

@ -401,8 +401,9 @@ fn compare_predicate_entailment<'tcx>(
// Finally, resolve all regions. This catches wily misuses of // Finally, resolve all regions. This catches wily misuses of
// lifetime parameters. // lifetime parameters.
let mut outlives_environment = OutlivesEnvironment::new(param_env); let mut outlives_environment = OutlivesEnvironment::builder(param_env);
outlives_environment.add_implied_bounds(infcx, wf_tys, impl_m_hir_id); outlives_environment.add_implied_bounds(infcx, wf_tys, impl_m_hir_id);
let outlives_environment = outlives_environment.build();
infcx.check_region_obligations_and_report_errors( infcx.check_region_obligations_and_report_errors(
impl_m.def_id.expect_local(), impl_m.def_id.expect_local(),
&outlives_environment, &outlives_environment,
@ -1513,8 +1514,9 @@ pub fn check_type_bounds<'tcx>(
// Finally, resolve all regions. This catches wily misuses of // Finally, resolve all regions. This catches wily misuses of
// lifetime parameters. // lifetime parameters.
let mut outlives_environment = OutlivesEnvironment::new(param_env); let mut outlives_environment = OutlivesEnvironment::builder(param_env);
outlives_environment.add_implied_bounds(&infcx, assumed_wf_types, impl_ty_hir_id); outlives_environment.add_implied_bounds(&infcx, assumed_wf_types, impl_ty_hir_id);
let outlives_environment = outlives_environment.build();
infcx.check_region_obligations_and_report_errors( infcx.check_region_obligations_and_report_errors(
impl_ty.def_id.expect_local(), impl_ty.def_id.expect_local(),
&outlives_environment, &outlives_environment,

View file

@ -1,7 +1,7 @@
use crate::outlives::outlives_bounds::InferCtxtExt as _; use crate::outlives::outlives_bounds::InferCtxtExt as _;
use rustc_data_structures::fx::FxHashSet; use rustc_data_structures::fx::FxHashSet;
use rustc_hir as hir; use rustc_hir as hir;
use rustc_infer::infer::outlives::env::OutlivesEnvironment; use rustc_infer::infer::outlives::env::OutlivesEnvironmentBuilder;
use rustc_infer::infer::InferCtxt; use rustc_infer::infer::InferCtxt;
use rustc_middle::ty::Ty; use rustc_middle::ty::Ty;
@ -14,7 +14,7 @@ pub(crate) trait OutlivesEnvironmentExt<'tcx> {
); );
} }
impl<'tcx> OutlivesEnvironmentExt<'tcx> for OutlivesEnvironment<'tcx> { impl<'tcx> OutlivesEnvironmentExt<'tcx> for OutlivesEnvironmentBuilder<'tcx> {
/// This method adds "implied bounds" into the outlives environment. /// This method adds "implied bounds" into the outlives environment.
/// Implied bounds are outlives relationships that we can deduce /// Implied bounds are outlives relationships that we can deduce
/// on the basis that certain types must be well-formed -- these are /// on the basis that certain types must be well-formed -- these are

View file

@ -104,8 +104,9 @@ pub(super) fn enter_wf_checking_ctxt<'tcx, F>(
return; return;
} }
let mut outlives_environment = OutlivesEnvironment::new(param_env); let mut outlives_environment = OutlivesEnvironment::builder(param_env);
outlives_environment.add_implied_bounds(infcx, assumed_wf_types, body_id); outlives_environment.add_implied_bounds(infcx, assumed_wf_types, body_id);
let outlives_environment = outlives_environment.build();
infcx.check_region_obligations_and_report_errors(body_def_id, &outlives_environment); infcx.check_region_obligations_and_report_errors(body_def_id, &outlives_environment);
}) })
} }
@ -694,8 +695,9 @@ fn resolve_regions_with_wf_tys<'tcx>(
// region constraints get added and solved there and we need to test each // region constraints get added and solved there and we need to test each
// call individually. // call individually.
tcx.infer_ctxt().enter(|infcx| { tcx.infer_ctxt().enter(|infcx| {
let mut outlives_environment = OutlivesEnvironment::new(param_env); let mut outlives_environment = OutlivesEnvironment::builder(param_env);
outlives_environment.add_implied_bounds(&infcx, wf_tys.clone(), id); outlives_environment.add_implied_bounds(&infcx, wf_tys.clone(), id);
let outlives_environment = outlives_environment.build();
let region_bound_pairs = outlives_environment.region_bound_pairs(); let region_bound_pairs = outlives_environment.region_bound_pairs();
add_constraints(&infcx, region_bound_pairs); add_constraints(&infcx, region_bound_pairs);

View file

@ -166,8 +166,7 @@ fn get_impl_substs<'tcx>(
return None; return None;
}; };
Some((impl1_substs, impl2_substs)) Some((impl1_substs, impl2_substs))
}) })}
}
/// Returns a list of all of the unconstrained subst of the given impl. /// Returns a list of all of the unconstrained subst of the given impl.
/// ///