Skip to content

Commit

Permalink
Auto merge of #47373 - wesleywiser:incr_cache_hashing, r=michaelwoeri…
Browse files Browse the repository at this point in the history
…ster

[Incremental] Cache hashes for AdDef and ty::Slice<T>

r? @michaelwoerister
  • Loading branch information
bors committed Jan 22, 2018
2 parents ae920dc + 45bd091 commit 47a8eb7
Show file tree
Hide file tree
Showing 2 changed files with 51 additions and 13 deletions.
25 changes: 23 additions & 2 deletions src/librustc/ich/impls_ty.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,11 @@
//! This module contains `HashStable` implementations for various data types
//! from rustc::ty in no particular order.
use ich::{StableHashingContext, NodeIdHashingMode};
use ich::{Fingerprint, StableHashingContext, NodeIdHashingMode};
use rustc_data_structures::fx::FxHashMap;
use rustc_data_structures::stable_hasher::{HashStable, ToStableHashKey,
StableHasher, StableHasherResult};
use std::cell::RefCell;
use std::hash as std_hash;
use std::mem;
use middle::region;
Expand All @@ -26,7 +28,26 @@ for &'gcx ty::Slice<T>
fn hash_stable<W: StableHasherResult>(&self,
hcx: &mut StableHashingContext<'gcx>,
hasher: &mut StableHasher<W>) {
(&self[..]).hash_stable(hcx, hasher);
thread_local! {
static CACHE: RefCell<FxHashMap<(usize, usize), Fingerprint>> =
RefCell::new(FxHashMap());
}

let hash = CACHE.with(|cache| {
let key = (self.as_ptr() as usize, self.len());
if let Some(&hash) = cache.borrow().get(&key) {
return hash;
}

let mut hasher = StableHasher::new();
(&self[..]).hash_stable(hcx, &mut hasher);

let hash: Fingerprint = hasher.finish();
cache.borrow_mut().insert(key, hash);
hash
});

hash.hash_stable(hcx, hasher);
}
}

Expand Down
39 changes: 28 additions & 11 deletions src/librustc/ty/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ use hir::def::{Def, CtorKind, ExportMap};
use hir::def_id::{CrateNum, DefId, DefIndex, LocalDefId, CRATE_DEF_INDEX, LOCAL_CRATE};
use hir::map::DefPathData;
use hir::svh::Svh;
use ich::Fingerprint;
use ich::StableHashingContext;
use middle::const_val::ConstVal;
use middle::lang_items::{FnTraitLangItem, FnMutTraitLangItem, FnOnceTraitLangItem};
Expand All @@ -37,6 +38,7 @@ use util::common::ErrorReported;
use util::nodemap::{NodeSet, DefIdMap, FxHashMap, FxHashSet};

use serialize::{self, Encodable, Encoder};
use std::cell::RefCell;
use std::collections::BTreeMap;
use std::cmp;
use std::fmt;
Expand Down Expand Up @@ -1476,17 +1478,32 @@ impl<'gcx> HashStable<StableHashingContext<'gcx>> for AdtDef {
fn hash_stable<W: StableHasherResult>(&self,
hcx: &mut StableHashingContext<'gcx>,
hasher: &mut StableHasher<W>) {
let ty::AdtDef {
did,
ref variants,
ref flags,
ref repr,
} = *self;

did.hash_stable(hcx, hasher);
variants.hash_stable(hcx, hasher);
flags.hash_stable(hcx, hasher);
repr.hash_stable(hcx, hasher);
thread_local! {
static CACHE: RefCell<FxHashMap<usize, Fingerprint>> =
RefCell::new(FxHashMap());
}

let hash: Fingerprint = CACHE.with(|cache| {
let addr = self as *const AdtDef as usize;
*cache.borrow_mut().entry(addr).or_insert_with(|| {
let ty::AdtDef {
did,
ref variants,
ref flags,
ref repr,
} = *self;

let mut hasher = StableHasher::new();
did.hash_stable(hcx, &mut hasher);
variants.hash_stable(hcx, &mut hasher);
flags.hash_stable(hcx, &mut hasher);
repr.hash_stable(hcx, &mut hasher);

hasher.finish()
})
});

hash.hash_stable(hcx, hasher);
}
}

Expand Down

0 comments on commit 47a8eb7

Please sign in to comment.