mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wrap LLVMSetMetadata

Browse Source
This commit is contained in:
Luv-Ray 2024-09-19 18:45:23 +08:00
parent b7c5656713
commit 632342a135
2 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
compiler/rustc_codegen_llvm/src/builder.rs
View File

@ -678,22 +678,20 @@ fn range_metadata(&mut self, load: &'ll Value, range: WrappingRange) {
unsafe {
let llty = self.cx.val_ty(load);
let v = [
let md = [
llvm::LLVMValueAsMetadata(self.cx.const_uint_big(llty, range.start)),
llvm::LLVMValueAsMetadata(self.cx.const_uint_big(llty, range.end.wrapping_add(1))),
];
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, v.as_ptr(), v.len());
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(load, llvm::MD_range as c_uint, md);
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, md.as_ptr(), md.len());
self.set_metadata(load, llvm::MD_range as c_uint, md);
}
}
fn nonnull_metadata(&mut self, load: &'ll Value) {
unsafe {
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, ptr::null(), 0);
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(load, llvm::MD_nonnull as c_uint, md);
self.set_metadata(load, llvm::MD_nonnull as c_uint, md);
}
}
@ -742,9 +740,8 @@ fn store_with_flags(
//
// [1]: https://llvm.org/docs/LangRef.html#store-instruction
let one = llvm::LLVMValueAsMetadata(self.cx.const_i32(1));
let node = llvm::LLVMMDNodeInContext2(self.cx.llcx, &one, 1);
let node = llvm::LLVMMetadataAsValue(&self.llcx, node);
llvm::LLVMSetMetadata(store, llvm::MD_nontemporal as c_uint, node);
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, &one, 1);
self.set_metadata(store, llvm::MD_nontemporal as c_uint, md);
}
}
store
@ -1209,8 +1206,7 @@ fn atomic_fence(
fn set_invariant_load(&mut self, load: &'ll Value) {
unsafe {
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, ptr::null(), 0);
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(load, llvm::MD_invariant_load as c_uint, md);
self.set_metadata(load, llvm::MD_invariant_load as c_uint, md);
}
}
@ -1339,26 +1335,23 @@ fn position_at_start(&mut self, llbb: &'ll BasicBlock) {
fn align_metadata(&mut self, load: &'ll Value, align: Align) {
unsafe {
let v = [llvm::LLVMValueAsMetadata(self.cx.const_u64(align.bytes()))];
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, v.as_ptr(), v.len());
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(load, llvm::MD_align as c_uint, md);
let md = [llvm::LLVMValueAsMetadata(self.cx.const_u64(align.bytes()))];
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, md.as_ptr(), md.len());
self.set_metadata(load, llvm::MD_align as c_uint, md);
}
}
fn noundef_metadata(&mut self, load: &'ll Value) {
unsafe {
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, ptr::null(), 0);
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(load, llvm::MD_noundef as c_uint, md);
self.set_metadata(load, llvm::MD_noundef as c_uint, md);
}
}
pub(crate) fn set_unpredictable(&mut self, inst: &'ll Value) {
unsafe {
let md = llvm::LLVMMDNodeInContext2(self.cx.llcx, ptr::null(), 0);
let md = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(inst, llvm::MD_unpredictable as c_uint, md);
self.set_metadata(inst, llvm::MD_unpredictable as c_uint, md);
}
}

11
compiler/rustc_codegen_llvm/src/context.rs
View File

@ -1,6 +1,6 @@
use std::borrow::Borrow;
use std::cell::{Cell, RefCell};
use std::ffi::CStr;
use std::ffi::{c_uint, CStr};
use std::str;
use rustc_codegen_ssa::base::{wants_msvc_seh, wants_wasm_eh};
@ -30,6 +30,7 @@
use crate::back::write::to_llvm_code_model;
use crate::callee::get_fn;
use crate::debuginfo::metadata::apply_vcall_visibility_metadata;
use crate::llvm::Metadata;
use crate::type_::Type;
use crate::value::Value;
use crate::{attributes, coverageinfo, debuginfo, llvm, llvm_util};
@ -585,6 +586,14 @@ pub(crate) fn create_used_variable_impl(&self, name: &'static CStr, values: &[&'
llvm::LLVMSetSection(g, c"llvm.metadata".as_ptr());
}
}
/// A wrapper for [`llvm::LLVMSetMetadata`], but it takes `Metadata` as a parameter instead of `Value`.
pub(crate) fn set_metadata<'a>(&self, val: &'a Value, kind_id: c_uint, md: &'a Metadata) {
unsafe {
let node = llvm::LLVMMetadataAsValue(&self.llcx, md);
llvm::LLVMSetMetadata(val, kind_id, node);
}
}
}
impl<'ll, 'tcx> MiscCodegenMethods<'tcx> for CodegenCx<'ll, 'tcx> {