avoid naming LLVM basic blocks when fewer_names is true

This commit is contained in:
Erik Desjardins 2024-03-15 15:53:49 -04:00
parent c5b571310d
commit 129b5e48f0
5 changed files with 36 additions and 15 deletions

View File

@ -1,6 +1,7 @@
use std::borrow::Cow;
use std::cell::Cell;
use std::convert::TryFrom;
use std::fmt::Display;
use std::ops::Deref;
use gccjit::{
@ -526,14 +527,14 @@ impl<'a, 'gcc, 'tcx> BuilderMethods<'a, 'tcx> for Builder<'a, 'gcc, 'tcx> {
self.block
}
fn append_block(cx: &'a CodegenCx<'gcc, 'tcx>, func: RValue<'gcc>, name: &str) -> Block<'gcc> {
fn append_block(cx: &'a CodegenCx<'gcc, 'tcx>, func: RValue<'gcc>, name: impl Display) -> Block<'gcc> {
let func = cx.rvalue_as_function(func);
func.new_block(name)
func.new_block(name.to_string())
}
fn append_sibling_block(&mut self, name: &str) -> Block<'gcc> {
fn append_sibling_block(&mut self, name: impl Display) -> Block<'gcc> {
let func = self.current_func();
func.new_block(name)
func.new_block(name.to_string())
}
fn switch_to_block(&mut self, block: Self::BasicBlock) {

View File

@ -26,6 +26,7 @@ use rustc_target::abi::{self, call::FnAbi, Align, Size, WrappingRange};
use rustc_target::spec::{HasTargetSpec, SanitizerSet, Target};
use smallvec::SmallVec;
use std::borrow::Cow;
use std::fmt::Display;
use std::iter;
use std::ops::Deref;
use std::ptr;
@ -153,14 +154,24 @@ impl<'a, 'll, 'tcx> BuilderMethods<'a, 'tcx> for Builder<'a, 'll, 'tcx> {
fn set_span(&mut self, _span: Span) {}
fn append_block(cx: &'a CodegenCx<'ll, 'tcx>, llfn: &'ll Value, name: &str) -> &'ll BasicBlock {
fn append_block(
cx: &'a CodegenCx<'ll, 'tcx>,
llfn: &'ll Value,
name: impl Display,
) -> &'ll BasicBlock {
unsafe {
let name = SmallCStr::new(name);
llvm::LLVMAppendBasicBlockInContext(cx.llcx, llfn, name.as_ptr())
let c_str_name;
let name_ptr = if cx.tcx.sess.fewer_names() {
const { c"".as_ptr().cast() }
} else {
c_str_name = SmallCStr::new(&name.to_string());
c_str_name.as_ptr()
};
llvm::LLVMAppendBasicBlockInContext(cx.llcx, llfn, name_ptr)
}
}
fn append_sibling_block(&mut self, name: &str) -> &'ll BasicBlock {
fn append_sibling_block(&mut self, name: impl Display) -> &'ll BasicBlock {
Self::append_block(self.cx, self.llfn(), name)
}

View File

@ -11,6 +11,7 @@
#![feature(exact_size_is_empty)]
#![feature(extern_types)]
#![feature(hash_raw_entry)]
#![feature(inline_const)]
#![feature(iter_intersperse)]
#![feature(let_chains)]
#![feature(impl_trait_in_assoc_type)]

View File

@ -83,8 +83,11 @@ impl<'a, 'tcx> TerminatorCodegenHelper<'tcx> {
// Cross-funclet jump - need a trampoline
debug_assert!(base::wants_new_eh_instructions(fx.cx.tcx().sess));
debug!("llbb_with_cleanup: creating cleanup trampoline for {:?}", target);
let name = &format!("{:?}_cleanup_trampoline_{:?}", self.bb, target);
let trampoline_llbb = Bx::append_block(fx.cx, fx.llfn, name);
let trampoline_llbb = Bx::append_block(
fx.cx,
fx.llfn,
format_args!("{:?}_cleanup_trampoline_{:?}", self.bb, target),
);
let mut trampoline_bx = Bx::build(fx.cx, trampoline_llbb);
trampoline_bx.cleanup_ret(self.funclet(fx).unwrap(), Some(lltarget));
trampoline_llbb
@ -1565,7 +1568,7 @@ impl<'a, 'tcx, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
fn landing_pad_for_uncached(&mut self, bb: mir::BasicBlock) -> Bx::BasicBlock {
let llbb = self.llbb(bb);
if base::wants_new_eh_instructions(self.cx.sess()) {
let cleanup_bb = Bx::append_block(self.cx, self.llfn, &format!("funclet_{bb:?}"));
let cleanup_bb = Bx::append_block(self.cx, self.llfn, format_args!("funclet_{bb:?}"));
let mut cleanup_bx = Bx::build(self.cx, cleanup_bb);
let funclet = cleanup_bx.cleanup_pad(None, &[]);
cleanup_bx.br(llbb);
@ -1688,8 +1691,7 @@ impl<'a, 'tcx, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
pub fn try_llbb(&mut self, bb: mir::BasicBlock) -> Option<Bx::BasicBlock> {
match self.cached_llbbs[bb] {
CachedLlbb::None => {
// FIXME(eddyb) only name the block if `fewer_names` is `false`.
let llbb = Bx::append_block(self.cx, self.llfn, &format!("{bb:?}"));
let llbb = Bx::append_block(self.cx, self.llfn, format_args!("{bb:?}"));
self.cached_llbbs[bb] = CachedLlbb::Some(llbb);
Some(llbb)
}

View File

@ -22,6 +22,8 @@ use rustc_target::abi::call::FnAbi;
use rustc_target::abi::{Abi, Align, Scalar, Size, WrappingRange};
use rustc_target::spec::HasTargetSpec;
use std::fmt::Display;
#[derive(Copy, Clone)]
pub enum OverflowOp {
Add,
@ -49,9 +51,13 @@ pub trait BuilderMethods<'a, 'tcx>:
fn set_span(&mut self, span: Span);
// FIXME(eddyb) replace uses of this with `append_sibling_block`.
fn append_block(cx: &'a Self::CodegenCx, llfn: Self::Function, name: &str) -> Self::BasicBlock;
fn append_block(
cx: &'a Self::CodegenCx,
llfn: Self::Function,
name: impl Display,
) -> Self::BasicBlock;
fn append_sibling_block(&mut self, name: &str) -> Self::BasicBlock;
fn append_sibling_block(&mut self, name: impl Display) -> Self::BasicBlock;
fn switch_to_block(&mut self, llbb: Self::BasicBlock);