Skip to content

Commit

Permalink
YJIT: Suppress unnecessary unsafe block (#7634)
Browse files Browse the repository at this point in the history
  • Loading branch information
nobu committed Mar 31, 2023
1 parent 1d19776 commit 9e678cd
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 14 deletions.
26 changes: 13 additions & 13 deletions yjit/src/core.rs
Expand Up @@ -1366,7 +1366,7 @@ pub fn limit_block_versions(blockid: BlockId, ctx: &Context) -> Context {
/// the potential to run blocks which are not ready.
unsafe fn add_block_version(blockref: BlockRef, cb: &CodeBlock) {
// SAFETY: caller ensures initialization
let block = unsafe { blockref.as_ref() };
let block = blockref.as_ref();

// Function entry blocks must have stack size 0
assert!(!(block.iseq_range.start == 0 && block.ctx.stack_size > 0));
Expand All @@ -1389,7 +1389,7 @@ unsafe fn add_block_version(blockref: BlockRef, cb: &CodeBlock) {
// Creating an unaligned pointer is well defined unlike in C.
let value_address: *const VALUE = value_address.cast();

let object = unsafe { value_address.read_unaligned() };
let object = value_address.read_unaligned();
obj_written!(iseq, object);
}

Expand Down Expand Up @@ -2696,17 +2696,17 @@ pub fn defer_compilation(
/// Block must be initialized and incoming/outgoing edges
/// must also point to initialized blocks.
unsafe fn remove_from_graph(blockref: BlockRef) {
let block = unsafe { blockref.as_ref() };
let block = blockref.as_ref();

// Remove this block from the predecessor's targets
for pred_branchref in block.incoming.0.take().iter() {
// Branch from the predecessor to us
let pred_branch = unsafe { pred_branchref.as_ref() };
let pred_branch = pred_branchref.as_ref();

// If this is us, nullify the target block
for target_idx in 0..pred_branch.targets.len() {
// SAFETY: no mutation inside unsafe
let target_is_us = unsafe {
let target_is_us = {
pred_branch.targets[target_idx]
.ref_unchecked()
.as_ref()
Expand All @@ -2723,18 +2723,18 @@ unsafe fn remove_from_graph(blockref: BlockRef) {

// For each outgoing branch
for out_branchref in block.outgoing.iter() {
let out_branch = unsafe { out_branchref.as_ref() };
let out_branch = out_branchref.as_ref();
// For each successor block
for out_target in out_branch.targets.iter() {
// SAFETY: copying out an Option<BlockRef>. No mutation.
let succ_block: Option<BlockRef> = unsafe {
let succ_block: Option<BlockRef> = {
out_target.ref_unchecked().as_ref().and_then(|target| target.get_block())
};

if let Some(succ_block) = succ_block {
// Remove outgoing branch from the successor's incoming list
// SAFETY: caller promises the block has valid outgoing edges.
let succ_block = unsafe { succ_block.as_ref() };
let succ_block = succ_block.as_ref();
// Temporarily move out of succ_block.incoming.
let succ_incoming = succ_block.incoming.0.take();
let mut succ_incoming = succ_incoming.into_vec();
Expand All @@ -2760,7 +2760,7 @@ unsafe fn remove_from_graph(blockref: BlockRef) {
pub unsafe fn free_block(blockref: BlockRef, graph_intact: bool) {
// Careful with order here.
// First, remove all pointers to the referent block
unsafe {
{
block_assumptions_free(blockref);

if graph_intact {
Expand All @@ -2769,21 +2769,21 @@ pub unsafe fn free_block(blockref: BlockRef, graph_intact: bool) {
}

// SAFETY: we should now have a unique pointer to the block
unsafe { dealloc_block(blockref) }
dealloc_block(blockref)
}

/// Deallocate a block and its outgoing branches. Blocks own their outgoing branches.
/// Caller must ensure that we have unique ownership for the referent block
unsafe fn dealloc_block(blockref: BlockRef) {
unsafe {
{
for outgoing in blockref.as_ref().outgoing.iter() {
// this Box::from_raw matches the Box::into_raw from PendingBranch::into_branch
mem::drop(Box::from_raw(outgoing.as_ptr()));
}
}

// Deallocate the referent Block
unsafe {
{
// this Box::from_raw matches the Box::into_raw from JITState::into_block
mem::drop(Box::from_raw(blockref.as_ptr()));
}
Expand Down Expand Up @@ -3008,7 +3008,7 @@ impl<T> RefUnchecked for Cell<T> {
// SAFETY: pointer is dereferenceable because it's from a &Cell.
// It's up to the caller to follow aliasing rules with the output
// reference.
unsafe { self.as_ptr().as_ref().unwrap() }
self.as_ptr().as_ref().unwrap()
}
}

Expand Down
2 changes: 1 addition & 1 deletion yjit/src/cruby.rs
Expand Up @@ -614,7 +614,7 @@ macro_rules! obj_written {
($old: expr, $young: expr) => {
let (old, young): (VALUE, VALUE) = ($old, $young);
let src_loc = $crate::cruby::src_loc!();
unsafe { rb_yjit_obj_written(old, young, src_loc.file.as_ptr(), src_loc.line) };
rb_yjit_obj_written(old, young, src_loc.file.as_ptr(), src_loc.line);
};
}
pub(crate) use obj_written;
Expand Down

0 comments on commit 9e678cd

Please sign in to comment.