1use std::cmp::Ordering;
2use std::ffi::c_uint;
3use std::ptr;
4
5use rustc_abi::{
6 Align, BackendRepr, ExternAbi, Float, HasDataLayout, Primitive, Size, WrappingRange,
7};
8use rustc_codegen_ssa::base::{compare_simd_types, wants_msvc_seh, wants_wasm_eh};
9use rustc_codegen_ssa::common::{IntPredicate, TypeKind};
10use rustc_codegen_ssa::errors::{ExpectedPointerMutability, InvalidMonomorphization};
11use rustc_codegen_ssa::mir::operand::{OperandRef, OperandValue};
12use rustc_codegen_ssa::mir::place::{PlaceRef, PlaceValue};
13use rustc_codegen_ssa::traits::*;
14use rustc_data_structures::assert_matches;
15use rustc_hir as hir;
16use rustc_hir::def_id::LOCAL_CRATE;
17use rustc_hir::find_attr;
18use rustc_middle::mir::BinOp;
19use rustc_middle::ty::layout::{FnAbiOf, HasTyCtxt, HasTypingEnv, LayoutOf};
20use rustc_middle::ty::offload_meta::OffloadMetadata;
21use rustc_middle::ty::{self, GenericArgsRef, Instance, SimdAlign, Ty, TyCtxt, TypingEnv};
22use rustc_middle::{bug, span_bug};
23use rustc_session::config::CrateType;
24use rustc_span::{Span, Symbol, sym};
25use rustc_symbol_mangling::{mangle_internal_symbol, symbol_name_for_instance_in_crate};
26use rustc_target::callconv::PassMode;
27use rustc_target::spec::Os;
28use tracing::debug;
29
30use crate::abi::FnAbiLlvmExt;
31use crate::builder::Builder;
32use crate::builder::autodiff::{adjust_activity_to_abi, generate_enzyme_call};
33use crate::builder::gpu_offload::{
34 OffloadKernelDims, gen_call_handling, gen_define_handling, register_offload,
35};
36use crate::context::CodegenCx;
37use crate::declare::declare_raw_fn;
38use crate::errors::{
39 AutoDiffWithoutEnable, AutoDiffWithoutLto, OffloadWithoutEnable, OffloadWithoutFatLTO,
40};
41use crate::llvm::{self, Type, Value};
42use crate::type_of::LayoutLlvmExt;
43use crate::va_arg::emit_va_arg;
44
45fn call_simple_intrinsic<'ll, 'tcx>(
46 bx: &mut Builder<'_, 'll, 'tcx>,
47 name: Symbol,
48 args: &[OperandRef<'tcx, &'ll Value>],
49) -> Option<&'ll Value> {
50 let (base_name, type_params): (&'static str, &[&'ll Type]) = match name {
51 sym::sqrtf16 => ("llvm.sqrt", &[bx.type_f16()]),
52 sym::sqrtf32 => ("llvm.sqrt", &[bx.type_f32()]),
53 sym::sqrtf64 => ("llvm.sqrt", &[bx.type_f64()]),
54 sym::sqrtf128 => ("llvm.sqrt", &[bx.type_f128()]),
55
56 sym::powif16 => ("llvm.powi", &[bx.type_f16(), bx.type_i32()]),
57 sym::powif32 => ("llvm.powi", &[bx.type_f32(), bx.type_i32()]),
58 sym::powif64 => ("llvm.powi", &[bx.type_f64(), bx.type_i32()]),
59 sym::powif128 => ("llvm.powi", &[bx.type_f128(), bx.type_i32()]),
60
61 sym::sinf16 => ("llvm.sin", &[bx.type_f16()]),
62 sym::sinf32 => ("llvm.sin", &[bx.type_f32()]),
63 sym::sinf64 => ("llvm.sin", &[bx.type_f64()]),
64 sym::sinf128 => ("llvm.sin", &[bx.type_f128()]),
65
66 sym::cosf16 => ("llvm.cos", &[bx.type_f16()]),
67 sym::cosf32 => ("llvm.cos", &[bx.type_f32()]),
68 sym::cosf64 => ("llvm.cos", &[bx.type_f64()]),
69 sym::cosf128 => ("llvm.cos", &[bx.type_f128()]),
70
71 sym::powf16 => ("llvm.pow", &[bx.type_f16()]),
72 sym::powf32 => ("llvm.pow", &[bx.type_f32()]),
73 sym::powf64 => ("llvm.pow", &[bx.type_f64()]),
74 sym::powf128 => ("llvm.pow", &[bx.type_f128()]),
75
76 sym::expf16 => ("llvm.exp", &[bx.type_f16()]),
77 sym::expf32 => ("llvm.exp", &[bx.type_f32()]),
78 sym::expf64 => ("llvm.exp", &[bx.type_f64()]),
79 sym::expf128 => ("llvm.exp", &[bx.type_f128()]),
80
81 sym::exp2f16 => ("llvm.exp2", &[bx.type_f16()]),
82 sym::exp2f32 => ("llvm.exp2", &[bx.type_f32()]),
83 sym::exp2f64 => ("llvm.exp2", &[bx.type_f64()]),
84 sym::exp2f128 => ("llvm.exp2", &[bx.type_f128()]),
85
86 sym::logf16 => ("llvm.log", &[bx.type_f16()]),
87 sym::logf32 => ("llvm.log", &[bx.type_f32()]),
88 sym::logf64 => ("llvm.log", &[bx.type_f64()]),
89 sym::logf128 => ("llvm.log", &[bx.type_f128()]),
90
91 sym::log10f16 => ("llvm.log10", &[bx.type_f16()]),
92 sym::log10f32 => ("llvm.log10", &[bx.type_f32()]),
93 sym::log10f64 => ("llvm.log10", &[bx.type_f64()]),
94 sym::log10f128 => ("llvm.log10", &[bx.type_f128()]),
95
96 sym::log2f16 => ("llvm.log2", &[bx.type_f16()]),
97 sym::log2f32 => ("llvm.log2", &[bx.type_f32()]),
98 sym::log2f64 => ("llvm.log2", &[bx.type_f64()]),
99 sym::log2f128 => ("llvm.log2", &[bx.type_f128()]),
100
101 sym::fmaf16 => ("llvm.fma", &[bx.type_f16()]),
102 sym::fmaf32 => ("llvm.fma", &[bx.type_f32()]),
103 sym::fmaf64 => ("llvm.fma", &[bx.type_f64()]),
104 sym::fmaf128 => ("llvm.fma", &[bx.type_f128()]),
105
106 sym::fmuladdf16 => ("llvm.fmuladd", &[bx.type_f16()]),
107 sym::fmuladdf32 => ("llvm.fmuladd", &[bx.type_f32()]),
108 sym::fmuladdf64 => ("llvm.fmuladd", &[bx.type_f64()]),
109 sym::fmuladdf128 => ("llvm.fmuladd", &[bx.type_f128()]),
110
111 sym::fabsf16 => ("llvm.fabs", &[bx.type_f16()]),
112 sym::fabsf32 => ("llvm.fabs", &[bx.type_f32()]),
113 sym::fabsf64 => ("llvm.fabs", &[bx.type_f64()]),
114 sym::fabsf128 => ("llvm.fabs", &[bx.type_f128()]),
115
116 sym::minnumf16 => ("llvm.minnum", &[bx.type_f16()]),
117 sym::minnumf32 => ("llvm.minnum", &[bx.type_f32()]),
118 sym::minnumf64 => ("llvm.minnum", &[bx.type_f64()]),
119 sym::minnumf128 => ("llvm.minnum", &[bx.type_f128()]),
120
121 sym::maxnumf16 => ("llvm.maxnum", &[bx.type_f16()]),
129 sym::maxnumf32 => ("llvm.maxnum", &[bx.type_f32()]),
130 sym::maxnumf64 => ("llvm.maxnum", &[bx.type_f64()]),
131 sym::maxnumf128 => ("llvm.maxnum", &[bx.type_f128()]),
132
133 sym::copysignf16 => ("llvm.copysign", &[bx.type_f16()]),
141 sym::copysignf32 => ("llvm.copysign", &[bx.type_f32()]),
142 sym::copysignf64 => ("llvm.copysign", &[bx.type_f64()]),
143 sym::copysignf128 => ("llvm.copysign", &[bx.type_f128()]),
144
145 sym::floorf16 => ("llvm.floor", &[bx.type_f16()]),
146 sym::floorf32 => ("llvm.floor", &[bx.type_f32()]),
147 sym::floorf64 => ("llvm.floor", &[bx.type_f64()]),
148 sym::floorf128 => ("llvm.floor", &[bx.type_f128()]),
149
150 sym::ceilf16 => ("llvm.ceil", &[bx.type_f16()]),
151 sym::ceilf32 => ("llvm.ceil", &[bx.type_f32()]),
152 sym::ceilf64 => ("llvm.ceil", &[bx.type_f64()]),
153 sym::ceilf128 => ("llvm.ceil", &[bx.type_f128()]),
154
155 sym::truncf16 => ("llvm.trunc", &[bx.type_f16()]),
156 sym::truncf32 => ("llvm.trunc", &[bx.type_f32()]),
157 sym::truncf64 => ("llvm.trunc", &[bx.type_f64()]),
158 sym::truncf128 => ("llvm.trunc", &[bx.type_f128()]),
159
160 sym::round_ties_even_f16 => ("llvm.rint", &[bx.type_f16()]),
165 sym::round_ties_even_f32 => ("llvm.rint", &[bx.type_f32()]),
166 sym::round_ties_even_f64 => ("llvm.rint", &[bx.type_f64()]),
167 sym::round_ties_even_f128 => ("llvm.rint", &[bx.type_f128()]),
168
169 sym::roundf16 => ("llvm.round", &[bx.type_f16()]),
170 sym::roundf32 => ("llvm.round", &[bx.type_f32()]),
171 sym::roundf64 => ("llvm.round", &[bx.type_f64()]),
172 sym::roundf128 => ("llvm.round", &[bx.type_f128()]),
173
174 _ => return None,
175 };
176 Some(bx.call_intrinsic(
177 base_name,
178 type_params,
179 &args.iter().map(|arg| arg.immediate()).collect::<Vec<_>>(),
180 ))
181}
182
183impl<'ll, 'tcx> IntrinsicCallBuilderMethods<'tcx> for Builder<'_, 'll, 'tcx> {
184 fn codegen_intrinsic_call(
185 &mut self,
186 instance: ty::Instance<'tcx>,
187 args: &[OperandRef<'tcx, &'ll Value>],
188 result: PlaceRef<'tcx, &'ll Value>,
189 span: Span,
190 ) -> Result<(), ty::Instance<'tcx>> {
191 let tcx = self.tcx;
192
193 let name = tcx.item_name(instance.def_id());
194 let fn_args = instance.args;
195
196 let simple = call_simple_intrinsic(self, name, args);
197 let llval = match name {
198 _ if simple.is_some() => simple.unwrap(),
199 sym::ptr_mask => {
200 let ptr = args[0].immediate();
201 self.call_intrinsic(
202 "llvm.ptrmask",
203 &[self.val_ty(ptr), self.type_isize()],
204 &[ptr, args[1].immediate()],
205 )
206 }
207 sym::autodiff => {
208 codegen_autodiff(self, tcx, instance, args, result);
209 return Ok(());
210 }
211 sym::offload => {
212 if tcx.sess.opts.unstable_opts.offload.is_empty() {
213 let _ = tcx.dcx().emit_almost_fatal(OffloadWithoutEnable);
214 }
215
216 if tcx.sess.lto() != rustc_session::config::Lto::Fat {
217 let _ = tcx.dcx().emit_almost_fatal(OffloadWithoutFatLTO);
218 }
219
220 codegen_offload(self, tcx, instance, args);
221 return Ok(());
222 }
223 sym::is_val_statically_known => {
224 if let OperandValue::Immediate(imm) = args[0].val {
225 self.call_intrinsic(
226 "llvm.is.constant",
227 &[args[0].layout.immediate_llvm_type(self.cx)],
228 &[imm],
229 )
230 } else {
231 self.const_bool(false)
232 }
233 }
234 sym::select_unpredictable => {
235 let cond = args[0].immediate();
236 match (&args[1].layout, &args[2].layout) {
(left_val, right_val) => {
if !(*left_val == *right_val) {
let kind = ::core::panicking::AssertKind::Eq;
::core::panicking::assert_failed(kind, &*left_val, &*right_val,
::core::option::Option::None);
}
}
};assert_eq!(args[1].layout, args[2].layout);
237 let select = |bx: &mut Self, true_val, false_val| {
238 let result = bx.select(cond, true_val, false_val);
239 bx.set_unpredictable(&result);
240 result
241 };
242 match (args[1].val, args[2].val) {
243 (OperandValue::Ref(true_val), OperandValue::Ref(false_val)) => {
244 if !true_val.llextra.is_none() {
::core::panicking::panic("assertion failed: true_val.llextra.is_none()")
};assert!(true_val.llextra.is_none());
245 if !false_val.llextra.is_none() {
::core::panicking::panic("assertion failed: false_val.llextra.is_none()")
};assert!(false_val.llextra.is_none());
246 match (&true_val.align, &false_val.align) {
(left_val, right_val) => {
if !(*left_val == *right_val) {
let kind = ::core::panicking::AssertKind::Eq;
::core::panicking::assert_failed(kind, &*left_val, &*right_val,
::core::option::Option::None);
}
}
};assert_eq!(true_val.align, false_val.align);
247 let ptr = select(self, true_val.llval, false_val.llval);
248 let selected =
249 OperandValue::Ref(PlaceValue::new_sized(ptr, true_val.align));
250 selected.store(self, result);
251 return Ok(());
252 }
253 (OperandValue::Immediate(_), OperandValue::Immediate(_))
254 | (OperandValue::Pair(_, _), OperandValue::Pair(_, _)) => {
255 let true_val = args[1].immediate_or_packed_pair(self);
256 let false_val = args[2].immediate_or_packed_pair(self);
257 select(self, true_val, false_val)
258 }
259 (OperandValue::ZeroSized, OperandValue::ZeroSized) => return Ok(()),
260 _ => ::rustc_middle::util::bug::span_bug_fmt(span,
format_args!("Incompatible OperandValue for select_unpredictable"))span_bug!(span, "Incompatible OperandValue for select_unpredictable"),
261 }
262 }
263 sym::catch_unwind => {
264 catch_unwind_intrinsic(
265 self,
266 args[0].immediate(),
267 args[1].immediate(),
268 args[2].immediate(),
269 result,
270 );
271 return Ok(());
272 }
273 sym::breakpoint => self.call_intrinsic("llvm.debugtrap", &[], &[]),
274 sym::va_arg => {
275 match result.layout.backend_repr {
276 BackendRepr::Scalar(scalar) => {
277 match scalar.primitive() {
278 Primitive::Int(..) => {
279 if self.cx().size_of(result.layout.ty).bytes() < 4 {
280 let promoted_result = emit_va_arg(self, args[0], tcx.types.i32);
285 self.trunc(promoted_result, result.layout.llvm_type(self))
286 } else {
287 emit_va_arg(self, args[0], result.layout.ty)
288 }
289 }
290 Primitive::Float(Float::F16) => {
291 ::rustc_middle::util::bug::bug_fmt(format_args!("the va_arg intrinsic does not work with `f16`"))bug!("the va_arg intrinsic does not work with `f16`")
292 }
293 Primitive::Float(Float::F64) | Primitive::Pointer(_) => {
294 emit_va_arg(self, args[0], result.layout.ty)
295 }
296 Primitive::Float(Float::F32) => {
298 ::rustc_middle::util::bug::bug_fmt(format_args!("the va_arg intrinsic does not work with `f32`"))bug!("the va_arg intrinsic does not work with `f32`")
299 }
300 Primitive::Float(Float::F128) => {
301 ::rustc_middle::util::bug::bug_fmt(format_args!("the va_arg intrinsic does not work with `f128`"))bug!("the va_arg intrinsic does not work with `f128`")
302 }
303 }
304 }
305 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("the va_arg intrinsic does not work with non-scalar types"))bug!("the va_arg intrinsic does not work with non-scalar types"),
306 }
307 }
308
309 sym::volatile_load | sym::unaligned_volatile_load => {
310 let ptr = args[0].immediate();
311 let load = self.volatile_load(result.layout.llvm_type(self), ptr);
312 let align = if name == sym::unaligned_volatile_load {
313 1
314 } else {
315 result.layout.align.bytes() as u32
316 };
317 unsafe {
318 llvm::LLVMSetAlignment(load, align);
319 }
320 if !result.layout.is_zst() {
321 self.store_to_place(load, result.val);
322 }
323 return Ok(());
324 }
325 sym::volatile_store => {
326 let dst = args[0].deref(self.cx());
327 args[1].val.volatile_store(self, dst);
328 return Ok(());
329 }
330 sym::unaligned_volatile_store => {
331 let dst = args[0].deref(self.cx());
332 args[1].val.unaligned_volatile_store(self, dst);
333 return Ok(());
334 }
335 sym::prefetch_read_data
336 | sym::prefetch_write_data
337 | sym::prefetch_read_instruction
338 | sym::prefetch_write_instruction => {
339 let (rw, cache_type) = match name {
340 sym::prefetch_read_data => (0, 1),
341 sym::prefetch_write_data => (1, 1),
342 sym::prefetch_read_instruction => (0, 0),
343 sym::prefetch_write_instruction => (1, 0),
344 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("impossible case reached"))bug!(),
345 };
346 let ptr = args[0].immediate();
347 let locality = fn_args.const_at(1).to_leaf().to_i32();
348 self.call_intrinsic(
349 "llvm.prefetch",
350 &[self.val_ty(ptr)],
351 &[
352 ptr,
353 self.const_i32(rw),
354 self.const_i32(locality),
355 self.const_i32(cache_type),
356 ],
357 )
358 }
359 sym::carrying_mul_add => {
360 let (size, signed) = fn_args.type_at(0).int_size_and_signed(self.tcx);
361
362 let wide_llty = self.type_ix(size.bits() * 2);
363 let args = args.as_array().unwrap();
364 let [a, b, c, d] = args.map(|a| self.intcast(a.immediate(), wide_llty, signed));
365
366 let wide = if signed {
367 let prod = self.unchecked_smul(a, b);
368 let acc = self.unchecked_sadd(prod, c);
369 self.unchecked_sadd(acc, d)
370 } else {
371 let prod = self.unchecked_umul(a, b);
372 let acc = self.unchecked_uadd(prod, c);
373 self.unchecked_uadd(acc, d)
374 };
375
376 let narrow_llty = self.type_ix(size.bits());
377 let low = self.trunc(wide, narrow_llty);
378 let bits_const = self.const_uint(wide_llty, size.bits());
379 let high = self.lshr(wide, bits_const);
381 let high = self.trunc(high, narrow_llty);
383
384 let pair_llty = self.type_struct(&[narrow_llty, narrow_llty], false);
385 let pair = self.const_poison(pair_llty);
386 let pair = self.insert_value(pair, low, 0);
387 let pair = self.insert_value(pair, high, 1);
388 pair
389 }
390
391 sym::carryless_mul if crate::llvm_util::get_version() >= (22, 0, 0) => {
393 let ty = args[0].layout.ty;
394 if !ty.is_integral() {
395 tcx.dcx().emit_err(InvalidMonomorphization::BasicIntegerType {
396 span,
397 name,
398 ty,
399 });
400 return Ok(());
401 }
402 let (size, _) = ty.int_size_and_signed(self.tcx);
403 let width = size.bits();
404 let llty = self.type_ix(width);
405
406 let lhs = args[0].immediate();
407 let rhs = args[1].immediate();
408 self.call_intrinsic("llvm.clmul", &[llty], &[lhs, rhs])
409 }
410
411 sym::ctlz
412 | sym::ctlz_nonzero
413 | sym::cttz
414 | sym::cttz_nonzero
415 | sym::ctpop
416 | sym::bswap
417 | sym::bitreverse
418 | sym::saturating_add
419 | sym::saturating_sub
420 | sym::unchecked_funnel_shl
421 | sym::unchecked_funnel_shr => {
422 let ty = args[0].layout.ty;
423 if !ty.is_integral() {
424 tcx.dcx().emit_err(InvalidMonomorphization::BasicIntegerType {
425 span,
426 name,
427 ty,
428 });
429 return Ok(());
430 }
431 let (size, signed) = ty.int_size_and_signed(self.tcx);
432 let width = size.bits();
433 let llty = self.type_ix(width);
434 match name {
435 sym::ctlz | sym::ctlz_nonzero | sym::cttz | sym::cttz_nonzero => {
436 let y =
437 self.const_bool(name == sym::ctlz_nonzero || name == sym::cttz_nonzero);
438 let llvm_name = if name == sym::ctlz || name == sym::ctlz_nonzero {
439 "llvm.ctlz"
440 } else {
441 "llvm.cttz"
442 };
443 let ret =
444 self.call_intrinsic(llvm_name, &[llty], &[args[0].immediate(), y]);
445 self.intcast(ret, result.layout.llvm_type(self), false)
446 }
447 sym::ctpop => {
448 let ret =
449 self.call_intrinsic("llvm.ctpop", &[llty], &[args[0].immediate()]);
450 self.intcast(ret, result.layout.llvm_type(self), false)
451 }
452 sym::bswap => {
453 if width == 8 {
454 args[0].immediate() } else {
456 self.call_intrinsic("llvm.bswap", &[llty], &[args[0].immediate()])
457 }
458 }
459 sym::bitreverse => {
460 self.call_intrinsic("llvm.bitreverse", &[llty], &[args[0].immediate()])
461 }
462 sym::unchecked_funnel_shl | sym::unchecked_funnel_shr => {
463 let is_left = name == sym::unchecked_funnel_shl;
464 let lhs = args[0].immediate();
465 let rhs = args[1].immediate();
466 let raw_shift = args[2].immediate();
467 let llvm_name = ::alloc::__export::must_use({
::alloc::fmt::format(format_args!("llvm.fsh{0}",
if is_left { 'l' } else { 'r' }))
})format!("llvm.fsh{}", if is_left { 'l' } else { 'r' });
468
469 let raw_shift = self.intcast(raw_shift, self.val_ty(lhs), false);
472
473 self.call_intrinsic(llvm_name, &[llty], &[lhs, rhs, raw_shift])
474 }
475 sym::saturating_add | sym::saturating_sub => {
476 let is_add = name == sym::saturating_add;
477 let lhs = args[0].immediate();
478 let rhs = args[1].immediate();
479 let llvm_name = ::alloc::__export::must_use({
::alloc::fmt::format(format_args!("llvm.{0}{1}.sat",
if signed { 's' } else { 'u' },
if is_add { "add" } else { "sub" }))
})format!(
480 "llvm.{}{}.sat",
481 if signed { 's' } else { 'u' },
482 if is_add { "add" } else { "sub" },
483 );
484 self.call_intrinsic(llvm_name, &[llty], &[lhs, rhs])
485 }
486 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("impossible case reached"))bug!(),
487 }
488 }
489
490 sym::raw_eq => {
491 use BackendRepr::*;
492 let tp_ty = fn_args.type_at(0);
493 let layout = self.layout_of(tp_ty).layout;
494 let use_integer_compare = match layout.backend_repr() {
495 Scalar(_) | ScalarPair(_, _) => true,
496 SimdVector { .. } => false,
497 ScalableVector { .. } => {
498 tcx.dcx().emit_err(InvalidMonomorphization::NonScalableType {
499 span,
500 name: sym::raw_eq,
501 ty: tp_ty,
502 });
503 return Ok(());
504 }
505 Memory { .. } => {
506 layout.size() <= self.data_layout().pointer_size() * 2
510 }
511 };
512
513 let a = args[0].immediate();
514 let b = args[1].immediate();
515 if layout.size().bytes() == 0 {
516 self.const_bool(true)
517 } else if use_integer_compare {
518 let integer_ty = self.type_ix(layout.size().bits());
519 let a_val = self.load(integer_ty, a, layout.align().abi);
520 let b_val = self.load(integer_ty, b, layout.align().abi);
521 self.icmp(IntPredicate::IntEQ, a_val, b_val)
522 } else {
523 let n = self.const_usize(layout.size().bytes());
524 let cmp = self.call_intrinsic("memcmp", &[], &[a, b, n]);
525 self.icmp(IntPredicate::IntEQ, cmp, self.const_int(self.type_int(), 0))
526 }
527 }
528
529 sym::compare_bytes => {
530 let cmp = self.call_intrinsic(
532 "memcmp",
533 &[],
534 &[args[0].immediate(), args[1].immediate(), args[2].immediate()],
535 );
536 self.sext(cmp, self.type_ix(32))
538 }
539
540 sym::black_box => {
541 args[0].val.store(self, result);
542 let result_val_span = [result.val.llval];
543 let (constraint, inputs): (&str, &[_]) = if result.layout.is_zst() {
553 ("~{memory}", &[])
554 } else {
555 ("r,~{memory}", &result_val_span)
556 };
557 crate::asm::inline_asm_call(
558 self,
559 "",
560 constraint,
561 inputs,
562 self.type_void(),
563 &[],
564 true,
565 false,
566 llvm::AsmDialect::Att,
567 &[span],
568 false,
569 None,
570 None,
571 )
572 .unwrap_or_else(|| ::rustc_middle::util::bug::bug_fmt(format_args!("failed to generate inline asm call for `black_box`"))bug!("failed to generate inline asm call for `black_box`"));
573
574 return Ok(());
576 }
577
578 sym::amdgpu_dispatch_ptr => {
579 let val = self.call_intrinsic("llvm.amdgcn.dispatch.ptr", &[], &[]);
580 self.pointercast(val, self.type_ptr())
582 }
583
584 _ if name.as_str().starts_with("simd_") => {
585 let mut loaded_args = Vec::new();
588 for arg in args {
589 loaded_args.push(
590 if arg.layout.ty.is_simd()
595 && let OperandValue::Ref(place) = arg.val
596 {
597 let (size, elem_ty) = arg.layout.ty.simd_size_and_type(self.tcx());
598 let elem_ll_ty = match elem_ty.kind() {
599 ty::Float(f) => self.type_float_from_ty(*f),
600 ty::Int(i) => self.type_int_from_ty(*i),
601 ty::Uint(u) => self.type_uint_from_ty(*u),
602 ty::RawPtr(_, _) => self.type_ptr(),
603 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
604 };
605 let loaded =
606 self.load_from_place(self.type_vector(elem_ll_ty, size), place);
607 OperandRef::from_immediate_or_packed_pair(self, loaded, arg.layout)
608 } else {
609 *arg
610 },
611 );
612 }
613
614 let llret_ty = if result.layout.ty.is_simd()
615 && let BackendRepr::Memory { .. } = result.layout.backend_repr
616 {
617 let (size, elem_ty) = result.layout.ty.simd_size_and_type(self.tcx());
618 let elem_ll_ty = match elem_ty.kind() {
619 ty::Float(f) => self.type_float_from_ty(*f),
620 ty::Int(i) => self.type_int_from_ty(*i),
621 ty::Uint(u) => self.type_uint_from_ty(*u),
622 ty::RawPtr(_, _) => self.type_ptr(),
623 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
624 };
625 self.type_vector(elem_ll_ty, size)
626 } else {
627 result.layout.llvm_type(self)
628 };
629
630 match generic_simd_intrinsic(
631 self,
632 name,
633 fn_args,
634 &loaded_args,
635 result.layout.ty,
636 llret_ty,
637 span,
638 ) {
639 Ok(llval) => llval,
640 Err(()) => return Ok(()),
643 }
644 }
645
646 _ => {
647 {
use ::tracing::__macro_support::Callsite as _;
static __CALLSITE: ::tracing::callsite::DefaultCallsite =
{
static META: ::tracing::Metadata<'static> =
{
::tracing_core::metadata::Metadata::new("event compiler/rustc_codegen_llvm/src/intrinsic.rs:647",
"rustc_codegen_llvm::intrinsic", ::tracing::Level::DEBUG,
::tracing_core::__macro_support::Option::Some("compiler/rustc_codegen_llvm/src/intrinsic.rs"),
::tracing_core::__macro_support::Option::Some(647u32),
::tracing_core::__macro_support::Option::Some("rustc_codegen_llvm::intrinsic"),
::tracing_core::field::FieldSet::new(&["message"],
::tracing_core::callsite::Identifier(&__CALLSITE)),
::tracing::metadata::Kind::EVENT)
};
::tracing::callsite::DefaultCallsite::new(&META)
};
let enabled =
::tracing::Level::DEBUG <= ::tracing::level_filters::STATIC_MAX_LEVEL
&&
::tracing::Level::DEBUG <=
::tracing::level_filters::LevelFilter::current() &&
{
let interest = __CALLSITE.interest();
!interest.is_never() &&
::tracing::__macro_support::__is_enabled(__CALLSITE.metadata(),
interest)
};
if enabled {
(|value_set: ::tracing::field::ValueSet|
{
let meta = __CALLSITE.metadata();
::tracing::Event::dispatch(meta, &value_set);
;
})({
#[allow(unused_imports)]
use ::tracing::field::{debug, display, Value};
let mut iter = __CALLSITE.metadata().fields().iter();
__CALLSITE.metadata().fields().value_set(&[(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&format_args!("unknown intrinsic \'{0}\' -- falling back to default body",
name) as &dyn Value))])
});
} else { ; }
};debug!("unknown intrinsic '{}' -- falling back to default body", name);
648 return Err(ty::Instance::new_raw(instance.def_id(), instance.args));
650 }
651 };
652
653 if result.layout.ty.is_bool() {
654 let val = self.from_immediate(llval);
655 self.store_to_place(val, result.val);
656 } else if !result.layout.ty.is_unit() {
657 self.store_to_place(llval, result.val);
658 }
659 Ok(())
660 }
661
662 fn codegen_llvm_intrinsic_call(
663 &mut self,
664 instance: ty::Instance<'tcx>,
665 args: &[OperandRef<'tcx, Self::Value>],
666 is_cleanup: bool,
667 ) -> Self::Value {
668 let tcx = self.tcx();
669
670 let fn_ty = instance.ty(tcx, self.typing_env());
671 let fn_sig = match *fn_ty.kind() {
672 ty::FnDef(def_id, args) => {
673 tcx.instantiate_bound_regions_with_erased(tcx.fn_sig(def_id).instantiate(tcx, args))
674 }
675 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
676 };
677 if !!fn_sig.c_variadic {
::core::panicking::panic("assertion failed: !fn_sig.c_variadic")
};assert!(!fn_sig.c_variadic);
678
679 let ret_layout = self.layout_of(fn_sig.output());
680 let llreturn_ty = if ret_layout.is_zst() {
681 self.type_void()
682 } else {
683 ret_layout.immediate_llvm_type(self)
684 };
685
686 let mut llargument_tys = Vec::with_capacity(fn_sig.inputs().len());
687 for &arg in fn_sig.inputs() {
688 let arg_layout = self.layout_of(arg);
689 if arg_layout.is_zst() {
690 continue;
691 }
692 llargument_tys.push(arg_layout.immediate_llvm_type(self));
693 }
694
695 let fn_ty = self.type_func(&llargument_tys, llreturn_ty);
696
697 let fn_ptr = if let Some(&llfn) = self.intrinsic_instances.borrow().get(&instance) {
698 llfn
699 } else {
700 let sym = tcx.symbol_name(instance).name;
701
702 let llfn = if let Some(llfn) = self.get_declared_value(sym) {
704 llfn
705 } else {
706 let llfn = declare_raw_fn(
709 self,
710 sym,
711 llvm::CCallConv,
712 llvm::UnnamedAddr::Global,
713 llvm::Visibility::Default,
714 fn_ty,
715 );
716
717 llfn
718 };
719
720 self.intrinsic_instances.borrow_mut().insert(instance, llfn);
721
722 llfn
723 };
724
725 let mut llargs = ::alloc::vec::Vec::new()vec![];
726
727 for arg in args {
728 match arg.val {
729 OperandValue::ZeroSized => {}
730 OperandValue::Immediate(_) => llargs.push(arg.immediate()),
731 OperandValue::Pair(a, b) => {
732 llargs.push(a);
733 llargs.push(b);
734 }
735 OperandValue::Ref(op_place_val) => {
736 let mut llval = op_place_val.llval;
737 llval = self.load(self.backend_type(arg.layout), llval, op_place_val.align);
743 if let BackendRepr::Scalar(scalar) = arg.layout.backend_repr {
744 if scalar.is_bool() {
745 self.range_metadata(llval, WrappingRange { start: 0, end: 1 });
746 }
747 llval = self.to_immediate_scalar(llval, scalar);
749 }
750 llargs.push(llval);
751 }
752 }
753 }
754
755 {
use ::tracing::__macro_support::Callsite as _;
static __CALLSITE: ::tracing::callsite::DefaultCallsite =
{
static META: ::tracing::Metadata<'static> =
{
::tracing_core::metadata::Metadata::new("event compiler/rustc_codegen_llvm/src/intrinsic.rs:755",
"rustc_codegen_llvm::intrinsic", ::tracing::Level::DEBUG,
::tracing_core::__macro_support::Option::Some("compiler/rustc_codegen_llvm/src/intrinsic.rs"),
::tracing_core::__macro_support::Option::Some(755u32),
::tracing_core::__macro_support::Option::Some("rustc_codegen_llvm::intrinsic"),
::tracing_core::field::FieldSet::new(&["message"],
::tracing_core::callsite::Identifier(&__CALLSITE)),
::tracing::metadata::Kind::EVENT)
};
::tracing::callsite::DefaultCallsite::new(&META)
};
let enabled =
::tracing::Level::DEBUG <= ::tracing::level_filters::STATIC_MAX_LEVEL
&&
::tracing::Level::DEBUG <=
::tracing::level_filters::LevelFilter::current() &&
{
let interest = __CALLSITE.interest();
!interest.is_never() &&
::tracing::__macro_support::__is_enabled(__CALLSITE.metadata(),
interest)
};
if enabled {
(|value_set: ::tracing::field::ValueSet|
{
let meta = __CALLSITE.metadata();
::tracing::Event::dispatch(meta, &value_set);
;
})({
#[allow(unused_imports)]
use ::tracing::field::{debug, display, Value};
let mut iter = __CALLSITE.metadata().fields().iter();
__CALLSITE.metadata().fields().value_set(&[(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&format_args!("call intrinsic {0:?} with args ({1:?})",
instance, llargs) as &dyn Value))])
});
} else { ; }
};debug!("call intrinsic {:?} with args ({:?})", instance, llargs);
756 let args = self.check_call("call", fn_ty, fn_ptr, &llargs);
757 let llret = unsafe {
758 llvm::LLVMBuildCallWithOperandBundles(
759 self.llbuilder,
760 fn_ty,
761 fn_ptr,
762 args.as_ptr() as *const &llvm::Value,
763 args.len() as c_uint,
764 ptr::dangling(),
765 0,
766 c"".as_ptr(),
767 )
768 };
769 if is_cleanup {
770 self.apply_attrs_to_cleanup_callsite(llret);
771 }
772
773 llret
774 }
775
776 fn abort(&mut self) {
777 self.call_intrinsic("llvm.trap", &[], &[]);
778 }
779
780 fn assume(&mut self, val: Self::Value) {
781 if self.cx.sess().opts.optimize != rustc_session::config::OptLevel::No {
782 self.call_intrinsic("llvm.assume", &[], &[val]);
783 }
784 }
785
786 fn expect(&mut self, cond: Self::Value, expected: bool) -> Self::Value {
787 if self.cx.sess().opts.optimize != rustc_session::config::OptLevel::No {
788 self.call_intrinsic(
789 "llvm.expect",
790 &[self.type_i1()],
791 &[cond, self.const_bool(expected)],
792 )
793 } else {
794 cond
795 }
796 }
797
798 fn type_checked_load(
799 &mut self,
800 llvtable: &'ll Value,
801 vtable_byte_offset: u64,
802 typeid: &[u8],
803 ) -> Self::Value {
804 let typeid = self.create_metadata(typeid);
805 let typeid = self.get_metadata_value(typeid);
806 let vtable_byte_offset = self.const_i32(vtable_byte_offset as i32);
807 let type_checked_load = self.call_intrinsic(
808 "llvm.type.checked.load",
809 &[],
810 &[llvtable, vtable_byte_offset, typeid],
811 );
812 self.extract_value(type_checked_load, 0)
813 }
814
815 fn va_start(&mut self, va_list: &'ll Value) -> &'ll Value {
816 self.call_intrinsic("llvm.va_start", &[self.val_ty(va_list)], &[va_list])
817 }
818
819 fn va_end(&mut self, va_list: &'ll Value) -> &'ll Value {
820 self.call_intrinsic("llvm.va_end", &[self.val_ty(va_list)], &[va_list])
821 }
822}
823
824fn catch_unwind_intrinsic<'ll, 'tcx>(
825 bx: &mut Builder<'_, 'll, 'tcx>,
826 try_func: &'ll Value,
827 data: &'ll Value,
828 catch_func: &'ll Value,
829 dest: PlaceRef<'tcx, &'ll Value>,
830) {
831 if !bx.sess().panic_strategy().unwinds() {
832 let try_func_ty = bx.type_func(&[bx.type_ptr()], bx.type_void());
833 bx.call(try_func_ty, None, None, try_func, &[data], None, None);
834 OperandValue::Immediate(bx.const_i32(0)).store(bx, dest);
837 } else if wants_msvc_seh(bx.sess()) {
838 codegen_msvc_try(bx, try_func, data, catch_func, dest);
839 } else if wants_wasm_eh(bx.sess()) {
840 codegen_wasm_try(bx, try_func, data, catch_func, dest);
841 } else if bx.sess().target.os == Os::Emscripten {
842 codegen_emcc_try(bx, try_func, data, catch_func, dest);
843 } else {
844 codegen_gnu_try(bx, try_func, data, catch_func, dest);
845 }
846}
847
848fn codegen_msvc_try<'ll, 'tcx>(
856 bx: &mut Builder<'_, 'll, 'tcx>,
857 try_func: &'ll Value,
858 data: &'ll Value,
859 catch_func: &'ll Value,
860 dest: PlaceRef<'tcx, &'ll Value>,
861) {
862 let (llty, llfn) = get_rust_try_fn(bx, &mut |mut bx| {
863 bx.set_personality_fn(bx.eh_personality());
864
865 let normal = bx.append_sibling_block("normal");
866 let catchswitch = bx.append_sibling_block("catchswitch");
867 let catchpad_rust = bx.append_sibling_block("catchpad_rust");
868 let catchpad_foreign = bx.append_sibling_block("catchpad_foreign");
869 let caught = bx.append_sibling_block("caught");
870
871 let try_func = llvm::get_param(bx.llfn(), 0);
872 let data = llvm::get_param(bx.llfn(), 1);
873 let catch_func = llvm::get_param(bx.llfn(), 2);
874
875 let ptr_size = bx.tcx().data_layout.pointer_size();
931 let ptr_align = bx.tcx().data_layout.pointer_align().abi;
932 let slot = bx.alloca(ptr_size, ptr_align);
933 let try_func_ty = bx.type_func(&[bx.type_ptr()], bx.type_void());
934 bx.invoke(try_func_ty, None, None, try_func, &[data], normal, catchswitch, None, None);
935
936 bx.switch_to_block(normal);
937 bx.ret(bx.const_i32(0));
938
939 bx.switch_to_block(catchswitch);
940 let cs = bx.catch_switch(None, None, &[catchpad_rust, catchpad_foreign]);
941
942 let type_info_vtable = bx.declare_global("??_7type_info@@6B@", bx.type_ptr());
957 let type_name = bx.const_bytes(b"rust_panic\0");
958 let type_info =
959 bx.const_struct(&[type_info_vtable, bx.const_null(bx.type_ptr()), type_name], false);
960 let tydesc = bx.declare_global(
961 &mangle_internal_symbol(bx.tcx, "__rust_panic_type_info"),
962 bx.val_ty(type_info),
963 );
964
965 llvm::set_linkage(tydesc, llvm::Linkage::LinkOnceODRLinkage);
966 if bx.cx.tcx.sess.target.supports_comdat() {
967 llvm::SetUniqueComdat(bx.llmod, tydesc);
968 }
969 llvm::set_initializer(tydesc, type_info);
970
971 bx.switch_to_block(catchpad_rust);
978 let flags = bx.const_i32(8);
979 let funclet = bx.catch_pad(cs, &[tydesc, flags, slot]);
980 let ptr = bx.load(bx.type_ptr(), slot, ptr_align);
981 let catch_ty = bx.type_func(&[bx.type_ptr(), bx.type_ptr()], bx.type_void());
982 bx.call(catch_ty, None, None, catch_func, &[data, ptr], Some(&funclet), None);
983 bx.catch_ret(&funclet, caught);
984
985 bx.switch_to_block(catchpad_foreign);
987 let flags = bx.const_i32(64);
988 let null = bx.const_null(bx.type_ptr());
989 let funclet = bx.catch_pad(cs, &[null, flags, null]);
990 bx.call(catch_ty, None, None, catch_func, &[data, null], Some(&funclet), None);
991 bx.catch_ret(&funclet, caught);
992
993 bx.switch_to_block(caught);
994 bx.ret(bx.const_i32(1));
995 });
996
997 let ret = bx.call(llty, None, None, llfn, &[try_func, data, catch_func], None, None);
1000 OperandValue::Immediate(ret).store(bx, dest);
1001}
1002
1003fn codegen_wasm_try<'ll, 'tcx>(
1005 bx: &mut Builder<'_, 'll, 'tcx>,
1006 try_func: &'ll Value,
1007 data: &'ll Value,
1008 catch_func: &'ll Value,
1009 dest: PlaceRef<'tcx, &'ll Value>,
1010) {
1011 let (llty, llfn) = get_rust_try_fn(bx, &mut |mut bx| {
1012 bx.set_personality_fn(bx.eh_personality());
1013
1014 let normal = bx.append_sibling_block("normal");
1015 let catchswitch = bx.append_sibling_block("catchswitch");
1016 let catchpad = bx.append_sibling_block("catchpad");
1017 let caught = bx.append_sibling_block("caught");
1018
1019 let try_func = llvm::get_param(bx.llfn(), 0);
1020 let data = llvm::get_param(bx.llfn(), 1);
1021 let catch_func = llvm::get_param(bx.llfn(), 2);
1022
1023 let try_func_ty = bx.type_func(&[bx.type_ptr()], bx.type_void());
1047 bx.invoke(try_func_ty, None, None, try_func, &[data], normal, catchswitch, None, None);
1048
1049 bx.switch_to_block(normal);
1050 bx.ret(bx.const_i32(0));
1051
1052 bx.switch_to_block(catchswitch);
1053 let cs = bx.catch_switch(None, None, &[catchpad]);
1054
1055 bx.switch_to_block(catchpad);
1056 let null = bx.const_null(bx.type_ptr());
1057 let funclet = bx.catch_pad(cs, &[null]);
1058
1059 let ptr = bx.call_intrinsic("llvm.wasm.get.exception", &[], &[funclet.cleanuppad()]);
1060 let _sel = bx.call_intrinsic("llvm.wasm.get.ehselector", &[], &[funclet.cleanuppad()]);
1061
1062 let catch_ty = bx.type_func(&[bx.type_ptr(), bx.type_ptr()], bx.type_void());
1063 bx.call(catch_ty, None, None, catch_func, &[data, ptr], Some(&funclet), None);
1064 bx.catch_ret(&funclet, caught);
1065
1066 bx.switch_to_block(caught);
1067 bx.ret(bx.const_i32(1));
1068 });
1069
1070 let ret = bx.call(llty, None, None, llfn, &[try_func, data, catch_func], None, None);
1073 OperandValue::Immediate(ret).store(bx, dest);
1074}
1075
1076fn codegen_gnu_try<'ll, 'tcx>(
1088 bx: &mut Builder<'_, 'll, 'tcx>,
1089 try_func: &'ll Value,
1090 data: &'ll Value,
1091 catch_func: &'ll Value,
1092 dest: PlaceRef<'tcx, &'ll Value>,
1093) {
1094 let (llty, llfn) = get_rust_try_fn(bx, &mut |mut bx| {
1095 let then = bx.append_sibling_block("then");
1108 let catch = bx.append_sibling_block("catch");
1109
1110 let try_func = llvm::get_param(bx.llfn(), 0);
1111 let data = llvm::get_param(bx.llfn(), 1);
1112 let catch_func = llvm::get_param(bx.llfn(), 2);
1113 let try_func_ty = bx.type_func(&[bx.type_ptr()], bx.type_void());
1114 bx.invoke(try_func_ty, None, None, try_func, &[data], then, catch, None, None);
1115
1116 bx.switch_to_block(then);
1117 bx.ret(bx.const_i32(0));
1118
1119 bx.switch_to_block(catch);
1126 let lpad_ty = bx.type_struct(&[bx.type_ptr(), bx.type_i32()], false);
1127 let vals = bx.landing_pad(lpad_ty, bx.eh_personality(), 1);
1128 let tydesc = bx.const_null(bx.type_ptr());
1129 bx.add_clause(vals, tydesc);
1130 let ptr = bx.extract_value(vals, 0);
1131 let catch_ty = bx.type_func(&[bx.type_ptr(), bx.type_ptr()], bx.type_void());
1132 bx.call(catch_ty, None, None, catch_func, &[data, ptr], None, None);
1133 bx.ret(bx.const_i32(1));
1134 });
1135
1136 let ret = bx.call(llty, None, None, llfn, &[try_func, data, catch_func], None, None);
1139 OperandValue::Immediate(ret).store(bx, dest);
1140}
1141
1142fn codegen_emcc_try<'ll, 'tcx>(
1146 bx: &mut Builder<'_, 'll, 'tcx>,
1147 try_func: &'ll Value,
1148 data: &'ll Value,
1149 catch_func: &'ll Value,
1150 dest: PlaceRef<'tcx, &'ll Value>,
1151) {
1152 let (llty, llfn) = get_rust_try_fn(bx, &mut |mut bx| {
1153 let then = bx.append_sibling_block("then");
1171 let catch = bx.append_sibling_block("catch");
1172
1173 let try_func = llvm::get_param(bx.llfn(), 0);
1174 let data = llvm::get_param(bx.llfn(), 1);
1175 let catch_func = llvm::get_param(bx.llfn(), 2);
1176 let try_func_ty = bx.type_func(&[bx.type_ptr()], bx.type_void());
1177 bx.invoke(try_func_ty, None, None, try_func, &[data], then, catch, None, None);
1178
1179 bx.switch_to_block(then);
1180 bx.ret(bx.const_i32(0));
1181
1182 bx.switch_to_block(catch);
1188 let tydesc = bx.eh_catch_typeinfo();
1189 let lpad_ty = bx.type_struct(&[bx.type_ptr(), bx.type_i32()], false);
1190 let vals = bx.landing_pad(lpad_ty, bx.eh_personality(), 2);
1191 bx.add_clause(vals, tydesc);
1192 bx.add_clause(vals, bx.const_null(bx.type_ptr()));
1193 let ptr = bx.extract_value(vals, 0);
1194 let selector = bx.extract_value(vals, 1);
1195
1196 let rust_typeid = bx.call_intrinsic("llvm.eh.typeid.for", &[bx.val_ty(tydesc)], &[tydesc]);
1198 let is_rust_panic = bx.icmp(IntPredicate::IntEQ, selector, rust_typeid);
1199 let is_rust_panic = bx.zext(is_rust_panic, bx.type_bool());
1200
1201 let ptr_size = bx.tcx().data_layout.pointer_size();
1204 let ptr_align = bx.tcx().data_layout.pointer_align().abi;
1205 let i8_align = bx.tcx().data_layout.i8_align;
1206 if !(i8_align <= ptr_align) {
::core::panicking::panic("assertion failed: i8_align <= ptr_align")
};assert!(i8_align <= ptr_align);
1208 let catch_data = bx.alloca(2 * ptr_size, ptr_align);
1209 bx.store(ptr, catch_data, ptr_align);
1210 let catch_data_1 = bx.inbounds_ptradd(catch_data, bx.const_usize(ptr_size.bytes()));
1211 bx.store(is_rust_panic, catch_data_1, i8_align);
1212
1213 let catch_ty = bx.type_func(&[bx.type_ptr(), bx.type_ptr()], bx.type_void());
1214 bx.call(catch_ty, None, None, catch_func, &[data, catch_data], None, None);
1215 bx.ret(bx.const_i32(1));
1216 });
1217
1218 let ret = bx.call(llty, None, None, llfn, &[try_func, data, catch_func], None, None);
1221 OperandValue::Immediate(ret).store(bx, dest);
1222}
1223
1224fn gen_fn<'a, 'll, 'tcx>(
1227 cx: &'a CodegenCx<'ll, 'tcx>,
1228 name: &str,
1229 rust_fn_sig: ty::PolyFnSig<'tcx>,
1230 codegen: &mut dyn FnMut(Builder<'a, 'll, 'tcx>),
1231) -> (&'ll Type, &'ll Value) {
1232 let fn_abi = cx.fn_abi_of_fn_ptr(rust_fn_sig, ty::List::empty());
1233 let llty = fn_abi.llvm_type(cx);
1234 let llfn = cx.declare_fn(name, fn_abi, None);
1235 cx.set_frame_pointer_type(llfn);
1236 cx.apply_target_cpu_attr(llfn);
1237 llvm::set_linkage(llfn, llvm::Linkage::InternalLinkage);
1239 let llbb = Builder::append_block(cx, llfn, "entry-block");
1240 let bx = Builder::build(cx, llbb);
1241 codegen(bx);
1242 (llty, llfn)
1243}
1244
1245fn get_rust_try_fn<'a, 'll, 'tcx>(
1250 cx: &'a CodegenCx<'ll, 'tcx>,
1251 codegen: &mut dyn FnMut(Builder<'a, 'll, 'tcx>),
1252) -> (&'ll Type, &'ll Value) {
1253 if let Some(llfn) = cx.rust_try_fn.get() {
1254 return llfn;
1255 }
1256
1257 let tcx = cx.tcx;
1259 let i8p = Ty::new_mut_ptr(tcx, tcx.types.i8);
1260 let try_fn_ty = Ty::new_fn_ptr(
1262 tcx,
1263 ty::Binder::dummy(tcx.mk_fn_sig(
1264 [i8p],
1265 tcx.types.unit,
1266 false,
1267 hir::Safety::Unsafe,
1268 ExternAbi::Rust,
1269 )),
1270 );
1271 let catch_fn_ty = Ty::new_fn_ptr(
1273 tcx,
1274 ty::Binder::dummy(tcx.mk_fn_sig(
1275 [i8p, i8p],
1276 tcx.types.unit,
1277 false,
1278 hir::Safety::Unsafe,
1279 ExternAbi::Rust,
1280 )),
1281 );
1282 let rust_fn_sig = ty::Binder::dummy(cx.tcx.mk_fn_sig(
1284 [try_fn_ty, i8p, catch_fn_ty],
1285 tcx.types.i32,
1286 false,
1287 hir::Safety::Unsafe,
1288 ExternAbi::Rust,
1289 ));
1290 let rust_try = gen_fn(cx, "__rust_try", rust_fn_sig, codegen);
1291 cx.rust_try_fn.set(Some(rust_try));
1292 rust_try
1293}
1294
1295fn codegen_autodiff<'ll, 'tcx>(
1296 bx: &mut Builder<'_, 'll, 'tcx>,
1297 tcx: TyCtxt<'tcx>,
1298 instance: ty::Instance<'tcx>,
1299 args: &[OperandRef<'tcx, &'ll Value>],
1300 result: PlaceRef<'tcx, &'ll Value>,
1301) {
1302 if !tcx.sess.opts.unstable_opts.autodiff.contains(&rustc_session::config::AutoDiff::Enable) {
1303 let _ = tcx.dcx().emit_almost_fatal(AutoDiffWithoutEnable);
1304 }
1305
1306 let ct = tcx.crate_types();
1307 let lto = tcx.sess.lto();
1308 if ct.len() == 1 && ct.contains(&CrateType::Executable) {
1309 if lto != rustc_session::config::Lto::Fat {
1310 let _ = tcx.dcx().emit_almost_fatal(AutoDiffWithoutLto);
1311 }
1312 } else {
1313 if lto != rustc_session::config::Lto::Fat && !tcx.sess.opts.cg.linker_plugin_lto.enabled() {
1314 let _ = tcx.dcx().emit_almost_fatal(AutoDiffWithoutLto);
1315 }
1316 }
1317
1318 let fn_args = instance.args;
1319 let callee_ty = instance.ty(tcx, bx.typing_env());
1320
1321 let sig = callee_ty.fn_sig(tcx).skip_binder();
1322
1323 let ret_ty = sig.output();
1324 let llret_ty = bx.layout_of(ret_ty).llvm_type(bx);
1325
1326 let (source_id, source_args) = match fn_args.into_type_list(tcx)[0].kind() {
1328 ty::FnDef(def_id, source_params) => (def_id, source_params),
1329 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("invalid autodiff intrinsic args"))bug!("invalid autodiff intrinsic args"),
1330 };
1331
1332 let fn_source = match Instance::try_resolve(tcx, bx.cx.typing_env(), *source_id, source_args) {
1333 Ok(Some(instance)) => instance,
1334 Ok(None) => ::rustc_middle::util::bug::bug_fmt(format_args!("could not resolve ({0:?}, {1:?}) to a specific autodiff instance",
source_id, source_args))bug!(
1335 "could not resolve ({:?}, {:?}) to a specific autodiff instance",
1336 source_id,
1337 source_args
1338 ),
1339 Err(_) => {
1340 return;
1342 }
1343 };
1344
1345 let source_symbol = symbol_name_for_instance_in_crate(tcx, fn_source.clone(), LOCAL_CRATE);
1346 let Some(fn_to_diff) = bx.cx.get_function(&source_symbol) else {
1347 ::rustc_middle::util::bug::bug_fmt(format_args!("could not find source function"))bug!("could not find source function")
1348 };
1349
1350 let (diff_id, diff_args) = match fn_args.into_type_list(tcx)[1].kind() {
1351 ty::FnDef(def_id, diff_args) => (def_id, diff_args),
1352 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("invalid args"))bug!("invalid args"),
1353 };
1354
1355 let fn_diff = match Instance::try_resolve(tcx, bx.cx.typing_env(), *diff_id, diff_args) {
1356 Ok(Some(instance)) => instance,
1357 Ok(None) => ::rustc_middle::util::bug::bug_fmt(format_args!("could not resolve ({0:?}, {1:?}) to a specific autodiff instance",
diff_id, diff_args))bug!(
1358 "could not resolve ({:?}, {:?}) to a specific autodiff instance",
1359 diff_id,
1360 diff_args
1361 ),
1362 Err(_) => {
1363 return;
1365 }
1366 };
1367
1368 let val_arr = get_args_from_tuple(bx, args[2], fn_diff);
1369 let diff_symbol = symbol_name_for_instance_in_crate(tcx, fn_diff.clone(), LOCAL_CRATE);
1370
1371 let Some(Some(mut diff_attrs)) =
1372 {
#[allow(deprecated)]
{
{
'done:
{
for i in tcx.get_all_attrs(fn_diff.def_id()) {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(RustcAutodiff(attr)) => {
break 'done Some(attr.clone());
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}
}
}find_attr!(tcx, fn_diff.def_id(), RustcAutodiff(attr) => attr.clone())
1373 else {
1374 ::rustc_middle::util::bug::bug_fmt(format_args!("could not find autodiff attrs"))bug!("could not find autodiff attrs")
1375 };
1376
1377 adjust_activity_to_abi(
1378 tcx,
1379 fn_source,
1380 TypingEnv::fully_monomorphized(),
1381 &mut diff_attrs.input_activity,
1382 );
1383
1384 let fnc_tree =
1385 rustc_middle::ty::fnc_typetrees(tcx, fn_source.ty(tcx, TypingEnv::fully_monomorphized()));
1386
1387 generate_enzyme_call(
1389 bx,
1390 bx.cx,
1391 fn_to_diff,
1392 &diff_symbol,
1393 llret_ty,
1394 &val_arr,
1395 &diff_attrs,
1396 result,
1397 fnc_tree,
1398 );
1399}
1400
1401fn codegen_offload<'ll, 'tcx>(
1406 bx: &mut Builder<'_, 'll, 'tcx>,
1407 tcx: TyCtxt<'tcx>,
1408 instance: ty::Instance<'tcx>,
1409 args: &[OperandRef<'tcx, &'ll Value>],
1410) {
1411 let cx = bx.cx;
1412 let fn_args = instance.args;
1413
1414 let (target_id, target_args) = match fn_args.into_type_list(tcx)[0].kind() {
1415 ty::FnDef(def_id, params) => (def_id, params),
1416 _ => ::rustc_middle::util::bug::bug_fmt(format_args!("invalid offload intrinsic arg"))bug!("invalid offload intrinsic arg"),
1417 };
1418
1419 let fn_target = match Instance::try_resolve(tcx, cx.typing_env(), *target_id, target_args) {
1420 Ok(Some(instance)) => instance,
1421 Ok(None) => ::rustc_middle::util::bug::bug_fmt(format_args!("could not resolve ({0:?}, {1:?}) to a specific offload instance",
target_id, target_args))bug!(
1422 "could not resolve ({:?}, {:?}) to a specific offload instance",
1423 target_id,
1424 target_args
1425 ),
1426 Err(_) => {
1427 return;
1429 }
1430 };
1431
1432 let offload_dims = OffloadKernelDims::from_operands(bx, &args[1], &args[2]);
1433 let args = get_args_from_tuple(bx, args[3], fn_target);
1434 let target_symbol = symbol_name_for_instance_in_crate(tcx, fn_target, LOCAL_CRATE);
1435
1436 let sig = tcx.fn_sig(fn_target.def_id()).skip_binder();
1437 let sig = tcx.instantiate_bound_regions_with_erased(sig);
1438 let inputs = sig.inputs();
1439
1440 let metadata = inputs.iter().map(|ty| OffloadMetadata::from_ty(tcx, *ty)).collect::<Vec<_>>();
1441
1442 let types = inputs.iter().map(|ty| cx.layout_of(*ty).llvm_type(cx)).collect::<Vec<_>>();
1443
1444 let offload_globals_ref = cx.offload_globals.borrow();
1445 let offload_globals = match offload_globals_ref.as_ref() {
1446 Some(globals) => globals,
1447 None => {
1448 return;
1450 }
1451 };
1452 register_offload(cx);
1453 let offload_data = gen_define_handling(&cx, &metadata, target_symbol, offload_globals);
1454 gen_call_handling(bx, &offload_data, &args, &types, &metadata, offload_globals, &offload_dims);
1455}
1456
1457fn get_args_from_tuple<'ll, 'tcx>(
1458 bx: &mut Builder<'_, 'll, 'tcx>,
1459 tuple_op: OperandRef<'tcx, &'ll Value>,
1460 fn_instance: Instance<'tcx>,
1461) -> Vec<&'ll Value> {
1462 let cx = bx.cx;
1463 let fn_abi = cx.fn_abi_of_instance(fn_instance, ty::List::empty());
1464
1465 match tuple_op.val {
1466 OperandValue::Immediate(val) => ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[val]))vec![val],
1467 OperandValue::Pair(v1, v2) => ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[v1, v2]))vec![v1, v2],
1468 OperandValue::Ref(ptr) => {
1469 let tuple_place = PlaceRef { val: ptr, layout: tuple_op.layout };
1470
1471 let mut result = Vec::with_capacity(fn_abi.args.len());
1472 let mut tuple_index = 0;
1473
1474 for arg in &fn_abi.args {
1475 match arg.mode {
1476 PassMode::Ignore => {}
1477 PassMode::Direct(_) | PassMode::Cast { .. } => {
1478 let field = tuple_place.project_field(bx, tuple_index);
1479 let llvm_ty = field.layout.llvm_type(bx.cx);
1480 let val = bx.load(llvm_ty, field.val.llval, field.val.align);
1481 result.push(val);
1482 tuple_index += 1;
1483 }
1484 PassMode::Pair(_, _) => {
1485 let field = tuple_place.project_field(bx, tuple_index);
1486 let llvm_ty = field.layout.llvm_type(bx.cx);
1487 let pair_val = bx.load(llvm_ty, field.val.llval, field.val.align);
1488 result.push(bx.extract_value(pair_val, 0));
1489 result.push(bx.extract_value(pair_val, 1));
1490 tuple_index += 1;
1491 }
1492 PassMode::Indirect { .. } => {
1493 let field = tuple_place.project_field(bx, tuple_index);
1494 result.push(field.val.llval);
1495 tuple_index += 1;
1496 }
1497 }
1498 }
1499
1500 result
1501 }
1502
1503 OperandValue::ZeroSized => ::alloc::vec::Vec::new()vec![],
1504 }
1505}
1506
1507fn generic_simd_intrinsic<'ll, 'tcx>(
1508 bx: &mut Builder<'_, 'll, 'tcx>,
1509 name: Symbol,
1510 fn_args: GenericArgsRef<'tcx>,
1511 args: &[OperandRef<'tcx, &'ll Value>],
1512 ret_ty: Ty<'tcx>,
1513 llret_ty: &'ll Type,
1514 span: Span,
1515) -> Result<&'ll Value, ()> {
1516 macro_rules! return_error {
1517 ($diag: expr) => {{
1518 bx.sess().dcx().emit_err($diag);
1519 return Err(());
1520 }};
1521 }
1522
1523 macro_rules! require {
1524 ($cond: expr, $diag: expr) => {
1525 if !$cond {
1526 return_error!($diag);
1527 }
1528 };
1529 }
1530
1531 macro_rules! require_simd {
1532 ($ty: expr, $variant:ident) => {{
1533 require!($ty.is_simd(), InvalidMonomorphization::$variant { span, name, ty: $ty });
1534 $ty.simd_size_and_type(bx.tcx())
1535 }};
1536 }
1537
1538 macro_rules! require_int_or_uint_ty {
1540 ($ty: expr, $diag: expr) => {
1541 match $ty {
1542 ty::Int(i) => {
1543 i.bit_width().unwrap_or_else(|| bx.data_layout().pointer_size().bits())
1544 }
1545 ty::Uint(i) => {
1546 i.bit_width().unwrap_or_else(|| bx.data_layout().pointer_size().bits())
1547 }
1548 _ => {
1549 return_error!($diag);
1550 }
1551 }
1552 };
1553 }
1554
1555 let llvm_version = crate::llvm_util::get_version();
1556
1557 fn vector_mask_to_bitmask<'a, 'll, 'tcx>(
1571 bx: &mut Builder<'a, 'll, 'tcx>,
1572 i_xn: &'ll Value,
1573 in_elem_bitwidth: u64,
1574 in_len: u64,
1575 ) -> &'ll Value {
1576 let shift_idx = bx.cx.const_int(bx.type_ix(in_elem_bitwidth), (in_elem_bitwidth - 1) as _);
1578 let shift_indices = ::alloc::vec::from_elem(shift_idx, in_len as _)vec![shift_idx; in_len as _];
1579 let i_xn_msb = bx.lshr(i_xn, bx.const_vector(shift_indices.as_slice()));
1580 bx.trunc(i_xn_msb, bx.type_vector(bx.type_i1(), in_len))
1582 }
1583
1584 if truecfg!(debug_assertions) {
1586 for arg in args {
1587 if arg.layout.ty.is_simd() {
1588 match arg.val {
OperandValue::Immediate(_) => {}
ref left_val => {
::core::panicking::assert_matches_failed(left_val,
"OperandValue::Immediate(_)", ::core::option::Option::None);
}
};assert_matches!(arg.val, OperandValue::Immediate(_));
1589 }
1590 }
1591 }
1592
1593 if name == sym::simd_select_bitmask {
1594 let (len, _) = {
if !args[1].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdArgument {
span,
name,
ty: args[1].layout.ty,
});
return Err(());
};
};
args[1].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[1].layout.ty, SimdArgument);
1595
1596 let expected_int_bits = len.max(8).next_power_of_two();
1597 let expected_bytes = len.div_ceil(8);
1598
1599 let mask_ty = args[0].layout.ty;
1600 let mask = match mask_ty.kind() {
1601 ty::Int(i) if i.bit_width() == Some(expected_int_bits) => args[0].immediate(),
1602 ty::Uint(i) if i.bit_width() == Some(expected_int_bits) => args[0].immediate(),
1603 ty::Array(elem, len)
1604 if #[allow(non_exhaustive_omitted_patterns)] match elem.kind() {
ty::Uint(ty::UintTy::U8) => true,
_ => false,
}matches!(elem.kind(), ty::Uint(ty::UintTy::U8))
1605 && len
1606 .try_to_target_usize(bx.tcx)
1607 .expect("expected monomorphic const in codegen")
1608 == expected_bytes =>
1609 {
1610 let place = PlaceRef::alloca(bx, args[0].layout);
1611 args[0].val.store(bx, place);
1612 let int_ty = bx.type_ix(expected_bytes * 8);
1613 bx.load(int_ty, place.val.llval, Align::ONE)
1614 }
1615 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::InvalidBitmask {
span,
name,
mask_ty,
expected_int_bits,
expected_bytes,
});
return Err(());
}return_error!(InvalidMonomorphization::InvalidBitmask {
1616 span,
1617 name,
1618 mask_ty,
1619 expected_int_bits,
1620 expected_bytes
1621 }),
1622 };
1623
1624 let i1 = bx.type_i1();
1625 let im = bx.type_ix(len);
1626 let i1xn = bx.type_vector(i1, len);
1627 let m_im = bx.trunc(mask, im);
1628 let m_i1s = bx.bitcast(m_im, i1xn);
1629 return Ok(bx.select(m_i1s, args[1].immediate(), args[2].immediate()));
1630 }
1631
1632 if name == sym::simd_splat {
1633 let (_out_len, out_ty) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
1634
1635 if !(args[0].layout.ty == out_ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedVectorElementType {
span,
name,
expected_element: out_ty,
vector_type: ret_ty,
});
return Err(());
};
};require!(
1636 args[0].layout.ty == out_ty,
1637 InvalidMonomorphization::ExpectedVectorElementType {
1638 span,
1639 name,
1640 expected_element: out_ty,
1641 vector_type: ret_ty,
1642 }
1643 );
1644
1645 let poison_vec = bx.const_poison(llret_ty);
1647 let idx0 = bx.const_i32(0);
1648 let v0 = bx.insert_element(poison_vec, args[0].immediate(), idx0);
1649
1650 let splat = bx.shuffle_vector(v0, poison_vec, bx.const_null(llret_ty));
1653
1654 return Ok(splat);
1655 }
1656
1657 let (in_len, in_elem) = {
if !args[0].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdInput {
span,
name,
ty: args[0].layout.ty,
});
return Err(());
};
};
args[0].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[0].layout.ty, SimdInput);
1659 let in_ty = args[0].layout.ty;
1660
1661 let comparison = match name {
1662 sym::simd_eq => Some(BinOp::Eq),
1663 sym::simd_ne => Some(BinOp::Ne),
1664 sym::simd_lt => Some(BinOp::Lt),
1665 sym::simd_le => Some(BinOp::Le),
1666 sym::simd_gt => Some(BinOp::Gt),
1667 sym::simd_ge => Some(BinOp::Ge),
1668 _ => None,
1669 };
1670
1671 if let Some(cmp_op) = comparison {
1672 let (out_len, out_ty) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
1673
1674 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLengthInputType {
span,
name,
in_len,
in_ty,
ret_ty,
out_len,
});
return Err(());
};
};require!(
1675 in_len == out_len,
1676 InvalidMonomorphization::ReturnLengthInputType {
1677 span,
1678 name,
1679 in_len,
1680 in_ty,
1681 ret_ty,
1682 out_len
1683 }
1684 );
1685 if !(bx.type_kind(bx.element_type(llret_ty)) == TypeKind::Integer) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnIntegerType {
span,
name,
ret_ty,
out_ty,
});
return Err(());
};
};require!(
1686 bx.type_kind(bx.element_type(llret_ty)) == TypeKind::Integer,
1687 InvalidMonomorphization::ReturnIntegerType { span, name, ret_ty, out_ty }
1688 );
1689
1690 return Ok(compare_simd_types(
1691 bx,
1692 args[0].immediate(),
1693 args[1].immediate(),
1694 in_elem,
1695 llret_ty,
1696 cmp_op,
1697 ));
1698 }
1699
1700 if name == sym::simd_shuffle_const_generic {
1701 let idx = fn_args[2].expect_const().to_branch();
1702 let n = idx.len() as u64;
1703
1704 let (out_len, out_ty) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
1705 if !(out_len == n) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLength {
span,
name,
in_len: n,
ret_ty,
out_len,
});
return Err(());
};
};require!(
1706 out_len == n,
1707 InvalidMonomorphization::ReturnLength { span, name, in_len: n, ret_ty, out_len }
1708 );
1709 if !(in_elem == out_ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnElement {
span,
name,
in_elem,
in_ty,
ret_ty,
out_ty,
});
return Err(());
};
};require!(
1710 in_elem == out_ty,
1711 InvalidMonomorphization::ReturnElement { span, name, in_elem, in_ty, ret_ty, out_ty }
1712 );
1713
1714 let total_len = in_len * 2;
1715
1716 let indices: Option<Vec<_>> = idx
1717 .iter()
1718 .enumerate()
1719 .map(|(arg_idx, val)| {
1720 let idx = val.to_leaf().to_i32();
1721 if idx >= i32::try_from(total_len).unwrap() {
1722 bx.sess().dcx().emit_err(InvalidMonomorphization::SimdIndexOutOfBounds {
1723 span,
1724 name,
1725 arg_idx: arg_idx as u64,
1726 total_len: total_len.into(),
1727 });
1728 None
1729 } else {
1730 Some(bx.const_i32(idx))
1731 }
1732 })
1733 .collect();
1734 let Some(indices) = indices else {
1735 return Ok(bx.const_null(llret_ty));
1736 };
1737
1738 return Ok(bx.shuffle_vector(
1739 args[0].immediate(),
1740 args[1].immediate(),
1741 bx.const_vector(&indices),
1742 ));
1743 }
1744
1745 if name == sym::simd_shuffle {
1746 let idx_ty = args[2].layout.ty;
1748 let n: u64 = if idx_ty.is_simd()
1749 && #[allow(non_exhaustive_omitted_patterns)] match idx_ty.simd_size_and_type(bx.cx.tcx).1.kind()
{
ty::Uint(ty::UintTy::U32) => true,
_ => false,
}matches!(idx_ty.simd_size_and_type(bx.cx.tcx).1.kind(), ty::Uint(ty::UintTy::U32))
1750 {
1751 idx_ty.simd_size_and_type(bx.cx.tcx).0
1752 } else {
1753 {
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdShuffle {
span,
name,
ty: idx_ty,
});
return Err(());
}return_error!(InvalidMonomorphization::SimdShuffle { span, name, ty: idx_ty })
1754 };
1755
1756 let (out_len, out_ty) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
1757 if !(out_len == n) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLength {
span,
name,
in_len: n,
ret_ty,
out_len,
});
return Err(());
};
};require!(
1758 out_len == n,
1759 InvalidMonomorphization::ReturnLength { span, name, in_len: n, ret_ty, out_len }
1760 );
1761 if !(in_elem == out_ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnElement {
span,
name,
in_elem,
in_ty,
ret_ty,
out_ty,
});
return Err(());
};
};require!(
1762 in_elem == out_ty,
1763 InvalidMonomorphization::ReturnElement { span, name, in_elem, in_ty, ret_ty, out_ty }
1764 );
1765
1766 let total_len = u128::from(in_len) * 2;
1767
1768 let indices = args[2].immediate();
1770 for i in 0..n {
1771 let val = bx.const_get_elt(indices, i as u64);
1772 let idx = bx
1773 .const_to_opt_u128(val, true)
1774 .unwrap_or_else(|| ::rustc_middle::util::bug::bug_fmt(format_args!("typeck should have already ensured that these are const"))bug!("typeck should have already ensured that these are const"));
1775 if idx >= total_len {
1776 {
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdIndexOutOfBounds {
span,
name,
arg_idx: i,
total_len,
});
return Err(());
};return_error!(InvalidMonomorphization::SimdIndexOutOfBounds {
1777 span,
1778 name,
1779 arg_idx: i,
1780 total_len,
1781 });
1782 }
1783 }
1784
1785 return Ok(bx.shuffle_vector(args[0].immediate(), args[1].immediate(), indices));
1786 }
1787
1788 if name == sym::simd_insert || name == sym::simd_insert_dyn {
1789 if !(in_elem == args[2].layout.ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::InsertedType {
span,
name,
in_elem,
in_ty,
out_ty: args[2].layout.ty,
});
return Err(());
};
};require!(
1790 in_elem == args[2].layout.ty,
1791 InvalidMonomorphization::InsertedType {
1792 span,
1793 name,
1794 in_elem,
1795 in_ty,
1796 out_ty: args[2].layout.ty
1797 }
1798 );
1799
1800 let index_imm = if name == sym::simd_insert {
1801 let idx = bx
1802 .const_to_opt_u128(args[1].immediate(), false)
1803 .expect("typeck should have ensure that this is a const");
1804 if idx >= in_len.into() {
1805 {
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdIndexOutOfBounds {
span,
name,
arg_idx: 1,
total_len: in_len.into(),
});
return Err(());
};return_error!(InvalidMonomorphization::SimdIndexOutOfBounds {
1806 span,
1807 name,
1808 arg_idx: 1,
1809 total_len: in_len.into(),
1810 });
1811 }
1812 bx.const_i32(idx as i32)
1813 } else {
1814 args[1].immediate()
1815 };
1816
1817 return Ok(bx.insert_element(args[0].immediate(), args[2].immediate(), index_imm));
1818 }
1819 if name == sym::simd_extract || name == sym::simd_extract_dyn {
1820 if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};require!(
1821 ret_ty == in_elem,
1822 InvalidMonomorphization::ReturnType { span, name, in_elem, in_ty, ret_ty }
1823 );
1824 let index_imm = if name == sym::simd_extract {
1825 let idx = bx
1826 .const_to_opt_u128(args[1].immediate(), false)
1827 .expect("typeck should have ensure that this is a const");
1828 if idx >= in_len.into() {
1829 {
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdIndexOutOfBounds {
span,
name,
arg_idx: 1,
total_len: in_len.into(),
});
return Err(());
};return_error!(InvalidMonomorphization::SimdIndexOutOfBounds {
1830 span,
1831 name,
1832 arg_idx: 1,
1833 total_len: in_len.into(),
1834 });
1835 }
1836 bx.const_i32(idx as i32)
1837 } else {
1838 args[1].immediate()
1839 };
1840
1841 return Ok(bx.extract_element(args[0].immediate(), index_imm));
1842 }
1843
1844 if name == sym::simd_select {
1845 let m_elem_ty = in_elem;
1846 let m_len = in_len;
1847 let (v_len, _) = {
if !args[1].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdArgument {
span,
name,
ty: args[1].layout.ty,
});
return Err(());
};
};
args[1].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[1].layout.ty, SimdArgument);
1848 if !(m_len == v_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MismatchedLengths {
span,
name,
m_len,
v_len,
});
return Err(());
};
};require!(
1849 m_len == v_len,
1850 InvalidMonomorphization::MismatchedLengths { span, name, m_len, v_len }
1851 );
1852
1853 let m_i1s = if args[1].layout.ty.is_scalable_vector() {
1854 match m_elem_ty.kind() {
1855 ty::Bool => {}
1856 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: m_elem_ty,
});
return Err(());
}return_error!(InvalidMonomorphization::MaskWrongElementType {
1857 span,
1858 name,
1859 ty: m_elem_ty
1860 }),
1861 };
1862 let i1 = bx.type_i1();
1863 let i1xn = bx.type_scalable_vector(i1, m_len as u64);
1864 bx.trunc(args[0].immediate(), i1xn)
1865 } else {
1866 let in_elem_bitwidth = match m_elem_ty.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: m_elem_ty,
});
return Err(());
};
}
}require_int_or_uint_ty!(
1867 m_elem_ty.kind(),
1868 InvalidMonomorphization::MaskWrongElementType { span, name, ty: m_elem_ty }
1869 );
1870 vector_mask_to_bitmask(bx, args[0].immediate(), in_elem_bitwidth, m_len)
1871 };
1872
1873 return Ok(bx.select(m_i1s, args[1].immediate(), args[2].immediate()));
1874 }
1875
1876 if name == sym::simd_bitmask {
1877 let expected_int_bits = in_len.max(8).next_power_of_two();
1886 let expected_bytes = in_len.div_ceil(8);
1887
1888 let in_elem_bitwidth = match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: in_elem,
});
return Err(());
};
}
}require_int_or_uint_ty!(
1890 in_elem.kind(),
1891 InvalidMonomorphization::MaskWrongElementType { span, name, ty: in_elem }
1892 );
1893
1894 let i1xn = vector_mask_to_bitmask(bx, args[0].immediate(), in_elem_bitwidth, in_len);
1895 let i_ = bx.bitcast(i1xn, bx.type_ix(in_len));
1897
1898 match ret_ty.kind() {
1899 ty::Uint(i) if i.bit_width() == Some(expected_int_bits) => {
1900 return Ok(bx.zext(i_, bx.type_ix(expected_int_bits)));
1902 }
1903 ty::Array(elem, len)
1904 if #[allow(non_exhaustive_omitted_patterns)] match elem.kind() {
ty::Uint(ty::UintTy::U8) => true,
_ => false,
}matches!(elem.kind(), ty::Uint(ty::UintTy::U8))
1905 && len
1906 .try_to_target_usize(bx.tcx)
1907 .expect("expected monomorphic const in codegen")
1908 == expected_bytes =>
1909 {
1910 let ze = bx.zext(i_, bx.type_ix(expected_bytes * 8));
1912
1913 let ptr = bx.alloca(Size::from_bytes(expected_bytes), Align::ONE);
1915 bx.store(ze, ptr, Align::ONE);
1916 let array_ty = bx.type_array(bx.type_i8(), expected_bytes);
1917 return Ok(bx.load(array_ty, ptr, Align::ONE));
1918 }
1919 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::CannotReturn {
span,
name,
ret_ty,
expected_int_bits,
expected_bytes,
});
return Err(());
}return_error!(InvalidMonomorphization::CannotReturn {
1920 span,
1921 name,
1922 ret_ty,
1923 expected_int_bits,
1924 expected_bytes
1925 }),
1926 }
1927 }
1928
1929 fn simd_simple_float_intrinsic<'ll, 'tcx>(
1930 name: Symbol,
1931 in_elem: Ty<'_>,
1932 in_ty: Ty<'_>,
1933 in_len: u64,
1934 bx: &mut Builder<'_, 'll, 'tcx>,
1935 span: Span,
1936 args: &[OperandRef<'tcx, &'ll Value>],
1937 ) -> Result<&'ll Value, ()> {
1938 macro_rules! return_error {
1939 ($diag: expr) => {{
1940 bx.sess().dcx().emit_err($diag);
1941 return Err(());
1942 }};
1943 }
1944
1945 let ty::Float(f) = in_elem.kind() else {
1946 {
bx.sess().dcx().emit_err(InvalidMonomorphization::FloatingPointType {
span,
name,
in_ty,
});
return Err(());
};return_error!(InvalidMonomorphization::FloatingPointType { span, name, in_ty });
1947 };
1948 let elem_ty = bx.cx.type_float_from_ty(*f);
1949
1950 let vec_ty = bx.type_vector(elem_ty, in_len);
1951
1952 let intr_name = match name {
1953 sym::simd_ceil => "llvm.ceil",
1954 sym::simd_fabs => "llvm.fabs",
1955 sym::simd_fcos => "llvm.cos",
1956 sym::simd_fexp2 => "llvm.exp2",
1957 sym::simd_fexp => "llvm.exp",
1958 sym::simd_flog10 => "llvm.log10",
1959 sym::simd_flog2 => "llvm.log2",
1960 sym::simd_flog => "llvm.log",
1961 sym::simd_floor => "llvm.floor",
1962 sym::simd_fma => "llvm.fma",
1963 sym::simd_relaxed_fma => "llvm.fmuladd",
1964 sym::simd_fsin => "llvm.sin",
1965 sym::simd_fsqrt => "llvm.sqrt",
1966 sym::simd_round => "llvm.round",
1967 sym::simd_round_ties_even => "llvm.rint",
1968 sym::simd_trunc => "llvm.trunc",
1969 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnrecognizedIntrinsic {
span,
name,
});
return Err(());
}return_error!(InvalidMonomorphization::UnrecognizedIntrinsic { span, name }),
1970 };
1971 Ok(bx.call_intrinsic(
1972 intr_name,
1973 &[vec_ty],
1974 &args.iter().map(|arg| arg.immediate()).collect::<Vec<_>>(),
1975 ))
1976 }
1977
1978 if #[allow(non_exhaustive_omitted_patterns)] match name {
sym::simd_ceil | sym::simd_fabs | sym::simd_fcos | sym::simd_fexp2 |
sym::simd_fexp | sym::simd_flog10 | sym::simd_flog2 | sym::simd_flog |
sym::simd_floor | sym::simd_fma | sym::simd_fsin | sym::simd_fsqrt |
sym::simd_relaxed_fma | sym::simd_round | sym::simd_round_ties_even |
sym::simd_trunc => true,
_ => false,
}std::matches!(
1979 name,
1980 sym::simd_ceil
1981 | sym::simd_fabs
1982 | sym::simd_fcos
1983 | sym::simd_fexp2
1984 | sym::simd_fexp
1985 | sym::simd_flog10
1986 | sym::simd_flog2
1987 | sym::simd_flog
1988 | sym::simd_floor
1989 | sym::simd_fma
1990 | sym::simd_fsin
1991 | sym::simd_fsqrt
1992 | sym::simd_relaxed_fma
1993 | sym::simd_round
1994 | sym::simd_round_ties_even
1995 | sym::simd_trunc
1996 ) {
1997 return simd_simple_float_intrinsic(name, in_elem, in_ty, in_len, bx, span, args);
1998 }
1999
2000 fn llvm_vector_ty<'ll>(cx: &CodegenCx<'ll, '_>, elem_ty: Ty<'_>, vec_len: u64) -> &'ll Type {
2001 let elem_ty = match *elem_ty.kind() {
2002 ty::Int(v) => cx.type_int_from_ty(v),
2003 ty::Uint(v) => cx.type_uint_from_ty(v),
2004 ty::Float(v) => cx.type_float_from_ty(v),
2005 ty::RawPtr(_, _) => cx.type_ptr(),
2006 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2007 };
2008 cx.type_vector(elem_ty, vec_len)
2009 }
2010
2011 if name == sym::simd_gather {
2012 let (_, element_ty0) = {
if !in_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdFirst {
span,
name,
ty: in_ty,
});
return Err(());
};
};
in_ty.simd_size_and_type(bx.tcx())
}require_simd!(in_ty, SimdFirst);
2023 let (out_len, element_ty1) = {
if !args[1].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdSecond {
span,
name,
ty: args[1].layout.ty,
});
return Err(());
};
};
args[1].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[1].layout.ty, SimdSecond);
2024 let (out_len2, element_ty2) = {
if !args[2].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdThird {
span,
name,
ty: args[2].layout.ty,
});
return Err(());
};
};
args[2].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[2].layout.ty, SimdThird);
2026 {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
};require_simd!(ret_ty, SimdReturn);
2027
2028 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SecondArgumentLength {
span,
name,
in_len,
in_ty,
arg_ty: args[1].layout.ty,
out_len,
});
return Err(());
};
};require!(
2030 in_len == out_len,
2031 InvalidMonomorphization::SecondArgumentLength {
2032 span,
2033 name,
2034 in_len,
2035 in_ty,
2036 arg_ty: args[1].layout.ty,
2037 out_len
2038 }
2039 );
2040 if !(in_len == out_len2) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ThirdArgumentLength {
span,
name,
in_len,
in_ty,
arg_ty: args[2].layout.ty,
out_len: out_len2,
});
return Err(());
};
};require!(
2041 in_len == out_len2,
2042 InvalidMonomorphization::ThirdArgumentLength {
2043 span,
2044 name,
2045 in_len,
2046 in_ty,
2047 arg_ty: args[2].layout.ty,
2048 out_len: out_len2
2049 }
2050 );
2051
2052 if !(ret_ty == in_ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedReturnType {
span,
name,
in_ty,
ret_ty,
});
return Err(());
};
};require!(
2054 ret_ty == in_ty,
2055 InvalidMonomorphization::ExpectedReturnType { span, name, in_ty, ret_ty }
2056 );
2057
2058 if !#[allow(non_exhaustive_omitted_patterns)] match *element_ty1.kind() {
ty::RawPtr(p_ty, _) if
p_ty == in_elem && p_ty.kind() == element_ty0.kind() => true,
_ => false,
} {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedElementType {
span,
name,
expected_element: element_ty1,
second_arg: args[1].layout.ty,
in_elem,
in_ty,
mutability: ExpectedPointerMutability::Not,
});
return Err(());
};
};require!(
2059 matches!(
2060 *element_ty1.kind(),
2061 ty::RawPtr(p_ty, _) if p_ty == in_elem && p_ty.kind() == element_ty0.kind()
2062 ),
2063 InvalidMonomorphization::ExpectedElementType {
2064 span,
2065 name,
2066 expected_element: element_ty1,
2067 second_arg: args[1].layout.ty,
2068 in_elem,
2069 in_ty,
2070 mutability: ExpectedPointerMutability::Not,
2071 }
2072 );
2073
2074 let mask_elem_bitwidth = match element_ty2.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: element_ty2,
});
return Err(());
};
}
}require_int_or_uint_ty!(
2075 element_ty2.kind(),
2076 InvalidMonomorphization::MaskWrongElementType { span, name, ty: element_ty2 }
2077 );
2078
2079 let alignment = bx.align_of(in_elem).bytes();
2081
2082 let mask = vector_mask_to_bitmask(bx, args[2].immediate(), mask_elem_bitwidth, in_len);
2084
2085 let llvm_pointer_vec_ty = llvm_vector_ty(bx, element_ty1, in_len);
2087
2088 let llvm_elem_vec_ty = llvm_vector_ty(bx, element_ty0, in_len);
2090
2091 let args: &[&'ll Value] = if llvm_version < (22, 0, 0) {
2092 let alignment = bx.const_i32(alignment as i32);
2093 &[args[1].immediate(), alignment, mask, args[0].immediate()]
2094 } else {
2095 &[args[1].immediate(), mask, args[0].immediate()]
2096 };
2097
2098 let call =
2099 bx.call_intrinsic("llvm.masked.gather", &[llvm_elem_vec_ty, llvm_pointer_vec_ty], args);
2100 if llvm_version >= (22, 0, 0) {
2101 crate::attributes::apply_to_callsite(
2102 call,
2103 crate::llvm::AttributePlace::Argument(0),
2104 &[crate::llvm::CreateAlignmentAttr(bx.llcx, alignment)],
2105 )
2106 }
2107 return Ok(call);
2108 }
2109
2110 fn llvm_alignment<'ll, 'tcx>(
2111 bx: &mut Builder<'_, 'll, 'tcx>,
2112 alignment: SimdAlign,
2113 vector_ty: Ty<'tcx>,
2114 element_ty: Ty<'tcx>,
2115 ) -> u64 {
2116 match alignment {
2117 SimdAlign::Unaligned => 1,
2118 SimdAlign::Element => bx.align_of(element_ty).bytes(),
2119 SimdAlign::Vector => bx.align_of(vector_ty).bytes(),
2120 }
2121 }
2122
2123 if name == sym::simd_masked_load {
2124 let alignment = fn_args[3].expect_const().to_branch()[0].to_leaf().to_simd_alignment();
2133
2134 let mask_ty = in_ty;
2136 let (mask_len, mask_elem) = (in_len, in_elem);
2137
2138 let pointer_ty = args[1].layout.ty;
2140
2141 let values_ty = args[2].layout.ty;
2143 let (values_len, values_elem) = {
if !values_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdThird {
span,
name,
ty: values_ty,
});
return Err(());
};
};
values_ty.simd_size_and_type(bx.tcx())
}require_simd!(values_ty, SimdThird);
2144
2145 {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
};require_simd!(ret_ty, SimdReturn);
2146
2147 if !(values_len == mask_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ThirdArgumentLength {
span,
name,
in_len: mask_len,
in_ty: mask_ty,
arg_ty: values_ty,
out_len: values_len,
});
return Err(());
};
};require!(
2149 values_len == mask_len,
2150 InvalidMonomorphization::ThirdArgumentLength {
2151 span,
2152 name,
2153 in_len: mask_len,
2154 in_ty: mask_ty,
2155 arg_ty: values_ty,
2156 out_len: values_len
2157 }
2158 );
2159
2160 if !(ret_ty == values_ty) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedReturnType {
span,
name,
in_ty: values_ty,
ret_ty,
});
return Err(());
};
};require!(
2162 ret_ty == values_ty,
2163 InvalidMonomorphization::ExpectedReturnType { span, name, in_ty: values_ty, ret_ty }
2164 );
2165
2166 if !#[allow(non_exhaustive_omitted_patterns)] match *pointer_ty.kind() {
ty::RawPtr(p_ty, _) if
p_ty == values_elem && p_ty.kind() == values_elem.kind() =>
true,
_ => false,
} {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedElementType {
span,
name,
expected_element: values_elem,
second_arg: pointer_ty,
in_elem: values_elem,
in_ty: values_ty,
mutability: ExpectedPointerMutability::Not,
});
return Err(());
};
};require!(
2167 matches!(
2168 *pointer_ty.kind(),
2169 ty::RawPtr(p_ty, _) if p_ty == values_elem && p_ty.kind() == values_elem.kind()
2170 ),
2171 InvalidMonomorphization::ExpectedElementType {
2172 span,
2173 name,
2174 expected_element: values_elem,
2175 second_arg: pointer_ty,
2176 in_elem: values_elem,
2177 in_ty: values_ty,
2178 mutability: ExpectedPointerMutability::Not,
2179 }
2180 );
2181
2182 let m_elem_bitwidth = match mask_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: mask_elem,
});
return Err(());
};
}
}require_int_or_uint_ty!(
2183 mask_elem.kind(),
2184 InvalidMonomorphization::MaskWrongElementType { span, name, ty: mask_elem }
2185 );
2186
2187 let mask = vector_mask_to_bitmask(bx, args[0].immediate(), m_elem_bitwidth, mask_len);
2188
2189 let alignment = llvm_alignment(bx, alignment, values_ty, values_elem);
2191
2192 let llvm_pointer = bx.type_ptr();
2193
2194 let llvm_elem_vec_ty = llvm_vector_ty(bx, values_elem, values_len);
2196
2197 let args: &[&'ll Value] = if llvm_version < (22, 0, 0) {
2198 let alignment = bx.const_i32(alignment as i32);
2199
2200 &[args[1].immediate(), alignment, mask, args[2].immediate()]
2201 } else {
2202 &[args[1].immediate(), mask, args[2].immediate()]
2203 };
2204
2205 let call = bx.call_intrinsic("llvm.masked.load", &[llvm_elem_vec_ty, llvm_pointer], args);
2206 if llvm_version >= (22, 0, 0) {
2207 crate::attributes::apply_to_callsite(
2208 call,
2209 crate::llvm::AttributePlace::Argument(0),
2210 &[crate::llvm::CreateAlignmentAttr(bx.llcx, alignment)],
2211 )
2212 }
2213 return Ok(call);
2214 }
2215
2216 if name == sym::simd_masked_store {
2217 let alignment = fn_args[3].expect_const().to_branch()[0].to_leaf().to_simd_alignment();
2226
2227 let mask_ty = in_ty;
2229 let (mask_len, mask_elem) = (in_len, in_elem);
2230
2231 let pointer_ty = args[1].layout.ty;
2233
2234 let values_ty = args[2].layout.ty;
2236 let (values_len, values_elem) = {
if !values_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdThird {
span,
name,
ty: values_ty,
});
return Err(());
};
};
values_ty.simd_size_and_type(bx.tcx())
}require_simd!(values_ty, SimdThird);
2237
2238 if !(values_len == mask_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ThirdArgumentLength {
span,
name,
in_len: mask_len,
in_ty: mask_ty,
arg_ty: values_ty,
out_len: values_len,
});
return Err(());
};
};require!(
2240 values_len == mask_len,
2241 InvalidMonomorphization::ThirdArgumentLength {
2242 span,
2243 name,
2244 in_len: mask_len,
2245 in_ty: mask_ty,
2246 arg_ty: values_ty,
2247 out_len: values_len
2248 }
2249 );
2250
2251 if !#[allow(non_exhaustive_omitted_patterns)] match *pointer_ty.kind() {
ty::RawPtr(p_ty, p_mutbl) if
p_ty == values_elem && p_ty.kind() == values_elem.kind() &&
p_mutbl.is_mut() => true,
_ => false,
} {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedElementType {
span,
name,
expected_element: values_elem,
second_arg: pointer_ty,
in_elem: values_elem,
in_ty: values_ty,
mutability: ExpectedPointerMutability::Mut,
});
return Err(());
};
};require!(
2253 matches!(
2254 *pointer_ty.kind(),
2255 ty::RawPtr(p_ty, p_mutbl)
2256 if p_ty == values_elem && p_ty.kind() == values_elem.kind() && p_mutbl.is_mut()
2257 ),
2258 InvalidMonomorphization::ExpectedElementType {
2259 span,
2260 name,
2261 expected_element: values_elem,
2262 second_arg: pointer_ty,
2263 in_elem: values_elem,
2264 in_ty: values_ty,
2265 mutability: ExpectedPointerMutability::Mut,
2266 }
2267 );
2268
2269 let m_elem_bitwidth = match mask_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: mask_elem,
});
return Err(());
};
}
}require_int_or_uint_ty!(
2270 mask_elem.kind(),
2271 InvalidMonomorphization::MaskWrongElementType { span, name, ty: mask_elem }
2272 );
2273
2274 let mask = vector_mask_to_bitmask(bx, args[0].immediate(), m_elem_bitwidth, mask_len);
2275
2276 let alignment = llvm_alignment(bx, alignment, values_ty, values_elem);
2278
2279 let llvm_pointer = bx.type_ptr();
2280
2281 let llvm_elem_vec_ty = llvm_vector_ty(bx, values_elem, values_len);
2283
2284 let args: &[&'ll Value] = if llvm_version < (22, 0, 0) {
2285 let alignment = bx.const_i32(alignment as i32);
2286 &[args[2].immediate(), args[1].immediate(), alignment, mask]
2287 } else {
2288 &[args[2].immediate(), args[1].immediate(), mask]
2289 };
2290
2291 let call = bx.call_intrinsic("llvm.masked.store", &[llvm_elem_vec_ty, llvm_pointer], args);
2292 if llvm_version >= (22, 0, 0) {
2293 crate::attributes::apply_to_callsite(
2294 call,
2295 crate::llvm::AttributePlace::Argument(1),
2296 &[crate::llvm::CreateAlignmentAttr(bx.llcx, alignment)],
2297 )
2298 }
2299 return Ok(call);
2300 }
2301
2302 if name == sym::simd_scatter {
2303 let (_, element_ty0) = {
if !in_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdFirst {
span,
name,
ty: in_ty,
});
return Err(());
};
};
in_ty.simd_size_and_type(bx.tcx())
}require_simd!(in_ty, SimdFirst);
2313 let (element_len1, element_ty1) = {
if !args[1].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdSecond {
span,
name,
ty: args[1].layout.ty,
});
return Err(());
};
};
args[1].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[1].layout.ty, SimdSecond);
2314 let (element_len2, element_ty2) = {
if !args[2].layout.ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdThird {
span,
name,
ty: args[2].layout.ty,
});
return Err(());
};
};
args[2].layout.ty.simd_size_and_type(bx.tcx())
}require_simd!(args[2].layout.ty, SimdThird);
2315
2316 if !(in_len == element_len1) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SecondArgumentLength {
span,
name,
in_len,
in_ty,
arg_ty: args[1].layout.ty,
out_len: element_len1,
});
return Err(());
};
};require!(
2318 in_len == element_len1,
2319 InvalidMonomorphization::SecondArgumentLength {
2320 span,
2321 name,
2322 in_len,
2323 in_ty,
2324 arg_ty: args[1].layout.ty,
2325 out_len: element_len1
2326 }
2327 );
2328 if !(in_len == element_len2) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ThirdArgumentLength {
span,
name,
in_len,
in_ty,
arg_ty: args[2].layout.ty,
out_len: element_len2,
});
return Err(());
};
};require!(
2329 in_len == element_len2,
2330 InvalidMonomorphization::ThirdArgumentLength {
2331 span,
2332 name,
2333 in_len,
2334 in_ty,
2335 arg_ty: args[2].layout.ty,
2336 out_len: element_len2
2337 }
2338 );
2339
2340 if !#[allow(non_exhaustive_omitted_patterns)] match *element_ty1.kind() {
ty::RawPtr(p_ty, p_mutbl) if
p_ty == in_elem && p_mutbl.is_mut() &&
p_ty.kind() == element_ty0.kind() => true,
_ => false,
} {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedElementType {
span,
name,
expected_element: element_ty1,
second_arg: args[1].layout.ty,
in_elem,
in_ty,
mutability: ExpectedPointerMutability::Mut,
});
return Err(());
};
};require!(
2341 matches!(
2342 *element_ty1.kind(),
2343 ty::RawPtr(p_ty, p_mutbl)
2344 if p_ty == in_elem && p_mutbl.is_mut() && p_ty.kind() == element_ty0.kind()
2345 ),
2346 InvalidMonomorphization::ExpectedElementType {
2347 span,
2348 name,
2349 expected_element: element_ty1,
2350 second_arg: args[1].layout.ty,
2351 in_elem,
2352 in_ty,
2353 mutability: ExpectedPointerMutability::Mut,
2354 }
2355 );
2356
2357 let mask_elem_bitwidth = match element_ty2.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::MaskWrongElementType {
span,
name,
ty: element_ty2,
});
return Err(());
};
}
}require_int_or_uint_ty!(
2359 element_ty2.kind(),
2360 InvalidMonomorphization::MaskWrongElementType { span, name, ty: element_ty2 }
2361 );
2362
2363 let alignment = bx.align_of(in_elem).bytes();
2365
2366 let mask = vector_mask_to_bitmask(bx, args[2].immediate(), mask_elem_bitwidth, in_len);
2368
2369 let llvm_pointer_vec_ty = llvm_vector_ty(bx, element_ty1, in_len);
2371
2372 let llvm_elem_vec_ty = llvm_vector_ty(bx, element_ty0, in_len);
2374 let args: &[&'ll Value] = if llvm_version < (22, 0, 0) {
2375 let alignment = bx.const_i32(alignment as i32);
2376 &[args[0].immediate(), args[1].immediate(), alignment, mask]
2377 } else {
2378 &[args[0].immediate(), args[1].immediate(), mask]
2379 };
2380 let call = bx.call_intrinsic(
2381 "llvm.masked.scatter",
2382 &[llvm_elem_vec_ty, llvm_pointer_vec_ty],
2383 args,
2384 );
2385 if llvm_version >= (22, 0, 0) {
2386 crate::attributes::apply_to_callsite(
2387 call,
2388 crate::llvm::AttributePlace::Argument(1),
2389 &[crate::llvm::CreateAlignmentAttr(bx.llcx, alignment)],
2390 )
2391 }
2392 return Ok(call);
2393 }
2394
2395 macro_rules! arith_red {
2396 ($name:ident : $integer_reduce:ident, $float_reduce:ident, $ordered:expr, $op:ident,
2397 $identity:expr) => {
2398 if name == sym::$name {
2399 require!(
2400 ret_ty == in_elem,
2401 InvalidMonomorphization::ReturnType { span, name, in_elem, in_ty, ret_ty }
2402 );
2403 return match in_elem.kind() {
2404 ty::Int(_) | ty::Uint(_) => {
2405 let r = bx.$integer_reduce(args[0].immediate());
2406 if $ordered {
2407 Ok(bx.$op(args[1].immediate(), r))
2410 } else {
2411 Ok(bx.$integer_reduce(args[0].immediate()))
2412 }
2413 }
2414 ty::Float(f) => {
2415 let acc = if $ordered {
2416 args[1].immediate()
2418 } else {
2419 match f.bit_width() {
2421 32 => bx.const_real(bx.type_f32(), $identity),
2422 64 => bx.const_real(bx.type_f64(), $identity),
2423 v => return_error!(
2424 InvalidMonomorphization::UnsupportedSymbolOfSize {
2425 span,
2426 name,
2427 symbol: sym::$name,
2428 in_ty,
2429 in_elem,
2430 size: v,
2431 ret_ty
2432 }
2433 ),
2434 }
2435 };
2436 Ok(bx.$float_reduce(acc, args[0].immediate()))
2437 }
2438 _ => return_error!(InvalidMonomorphization::UnsupportedSymbol {
2439 span,
2440 name,
2441 symbol: sym::$name,
2442 in_ty,
2443 in_elem,
2444 ret_ty
2445 }),
2446 };
2447 }
2448 };
2449 }
2450
2451 if name == sym::simd_reduce_add_ordered {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_add(args[0].immediate());
if true {
Ok(bx.add(args[1].immediate(), r))
} else { Ok(bx.vector_reduce_add(args[0].immediate())) }
}
ty::Float(f) => {
let acc =
if true {
args[1].immediate()
} else {
match f.bit_width() {
32 => bx.const_real(bx.type_f32(), -0.0),
64 => bx.const_real(bx.type_f64(), -0.0),
v => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbolOfSize {
span,
name,
symbol: sym::simd_reduce_add_ordered,
in_ty,
in_elem,
size: v,
ret_ty,
});
return Err(());
}
}
};
Ok(bx.vector_reduce_fadd(acc, args[0].immediate()))
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_add_ordered,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};arith_red!(simd_reduce_add_ordered: vector_reduce_add, vector_reduce_fadd, true, add, -0.0);
2452 if name == sym::simd_reduce_mul_ordered {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_mul(args[0].immediate());
if true {
Ok(bx.mul(args[1].immediate(), r))
} else { Ok(bx.vector_reduce_mul(args[0].immediate())) }
}
ty::Float(f) => {
let acc =
if true {
args[1].immediate()
} else {
match f.bit_width() {
32 => bx.const_real(bx.type_f32(), 1.0),
64 => bx.const_real(bx.type_f64(), 1.0),
v => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbolOfSize {
span,
name,
symbol: sym::simd_reduce_mul_ordered,
in_ty,
in_elem,
size: v,
ret_ty,
});
return Err(());
}
}
};
Ok(bx.vector_reduce_fmul(acc, args[0].immediate()))
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_mul_ordered,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};arith_red!(simd_reduce_mul_ordered: vector_reduce_mul, vector_reduce_fmul, true, mul, 1.0);
2453 if name == sym::simd_reduce_add_unordered {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_add(args[0].immediate());
if false {
Ok(bx.add(args[1].immediate(), r))
} else { Ok(bx.vector_reduce_add(args[0].immediate())) }
}
ty::Float(f) => {
let acc =
if false {
args[1].immediate()
} else {
match f.bit_width() {
32 => bx.const_real(bx.type_f32(), -0.0),
64 => bx.const_real(bx.type_f64(), -0.0),
v => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbolOfSize {
span,
name,
symbol: sym::simd_reduce_add_unordered,
in_ty,
in_elem,
size: v,
ret_ty,
});
return Err(());
}
}
};
Ok(bx.vector_reduce_fadd_reassoc(acc, args[0].immediate()))
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_add_unordered,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};arith_red!(
2454 simd_reduce_add_unordered: vector_reduce_add,
2455 vector_reduce_fadd_reassoc,
2456 false,
2457 add,
2458 -0.0
2459 );
2460 if name == sym::simd_reduce_mul_unordered {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_mul(args[0].immediate());
if false {
Ok(bx.mul(args[1].immediate(), r))
} else { Ok(bx.vector_reduce_mul(args[0].immediate())) }
}
ty::Float(f) => {
let acc =
if false {
args[1].immediate()
} else {
match f.bit_width() {
32 => bx.const_real(bx.type_f32(), 1.0),
64 => bx.const_real(bx.type_f64(), 1.0),
v => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbolOfSize {
span,
name,
symbol: sym::simd_reduce_mul_unordered,
in_ty,
in_elem,
size: v,
ret_ty,
});
return Err(());
}
}
};
Ok(bx.vector_reduce_fmul_reassoc(acc, args[0].immediate()))
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_mul_unordered,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};arith_red!(
2461 simd_reduce_mul_unordered: vector_reduce_mul,
2462 vector_reduce_fmul_reassoc,
2463 false,
2464 mul,
2465 1.0
2466 );
2467
2468 macro_rules! minmax_red {
2469 ($name:ident: $int_red:ident, $float_red:ident) => {
2470 if name == sym::$name {
2471 require!(
2472 ret_ty == in_elem,
2473 InvalidMonomorphization::ReturnType { span, name, in_elem, in_ty, ret_ty }
2474 );
2475 return match in_elem.kind() {
2476 ty::Int(_i) => Ok(bx.$int_red(args[0].immediate(), true)),
2477 ty::Uint(_u) => Ok(bx.$int_red(args[0].immediate(), false)),
2478 ty::Float(_f) => Ok(bx.$float_red(args[0].immediate())),
2479 _ => return_error!(InvalidMonomorphization::UnsupportedSymbol {
2480 span,
2481 name,
2482 symbol: sym::$name,
2483 in_ty,
2484 in_elem,
2485 ret_ty
2486 }),
2487 };
2488 }
2489 };
2490 }
2491
2492 if name == sym::simd_reduce_min {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_i) =>
Ok(bx.vector_reduce_min(args[0].immediate(), true)),
ty::Uint(_u) =>
Ok(bx.vector_reduce_min(args[0].immediate(), false)),
ty::Float(_f) => Ok(bx.vector_reduce_fmin(args[0].immediate())),
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_min,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};minmax_red!(simd_reduce_min: vector_reduce_min, vector_reduce_fmin);
2493 if name == sym::simd_reduce_max {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
return match in_elem.kind() {
ty::Int(_i) =>
Ok(bx.vector_reduce_max(args[0].immediate(), true)),
ty::Uint(_u) =>
Ok(bx.vector_reduce_max(args[0].immediate(), false)),
ty::Float(_f) => Ok(bx.vector_reduce_fmax(args[0].immediate())),
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_max,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};minmax_red!(simd_reduce_max: vector_reduce_max, vector_reduce_fmax);
2494
2495 macro_rules! bitwise_red {
2496 ($name:ident : $red:ident, $boolean:expr) => {
2497 if name == sym::$name {
2498 let input = if !$boolean {
2499 require!(
2500 ret_ty == in_elem,
2501 InvalidMonomorphization::ReturnType { span, name, in_elem, in_ty, ret_ty }
2502 );
2503 args[0].immediate()
2504 } else {
2505 let bitwidth = match in_elem.kind() {
2506 ty::Int(i) => {
2507 i.bit_width().unwrap_or_else(|| bx.data_layout().pointer_size().bits())
2508 }
2509 ty::Uint(i) => {
2510 i.bit_width().unwrap_or_else(|| bx.data_layout().pointer_size().bits())
2511 }
2512 _ => return_error!(InvalidMonomorphization::UnsupportedSymbol {
2513 span,
2514 name,
2515 symbol: sym::$name,
2516 in_ty,
2517 in_elem,
2518 ret_ty
2519 }),
2520 };
2521
2522 vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth, in_len as _)
2523 };
2524 return match in_elem.kind() {
2525 ty::Int(_) | ty::Uint(_) => {
2526 let r = bx.$red(input);
2527 Ok(if !$boolean { r } else { bx.zext(r, bx.type_bool()) })
2528 }
2529 _ => return_error!(InvalidMonomorphization::UnsupportedSymbol {
2530 span,
2531 name,
2532 symbol: sym::$name,
2533 in_ty,
2534 in_elem,
2535 ret_ty
2536 }),
2537 };
2538 }
2539 };
2540 }
2541
2542 if name == sym::simd_reduce_and {
let input =
if !false {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
args[0].immediate()
} else {
let bitwidth =
match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_and,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth,
in_len as _)
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_and(input);
Ok(if !false { r } else { bx.zext(r, bx.type_bool()) })
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_and,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};bitwise_red!(simd_reduce_and: vector_reduce_and, false);
2543 if name == sym::simd_reduce_or {
let input =
if !false {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
args[0].immediate()
} else {
let bitwidth =
match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_or,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth,
in_len as _)
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_or(input);
Ok(if !false { r } else { bx.zext(r, bx.type_bool()) })
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_or,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};bitwise_red!(simd_reduce_or: vector_reduce_or, false);
2544 if name == sym::simd_reduce_xor {
let input =
if !false {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
args[0].immediate()
} else {
let bitwidth =
match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_xor,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth,
in_len as _)
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_xor(input);
Ok(if !false { r } else { bx.zext(r, bx.type_bool()) })
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_xor,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};bitwise_red!(simd_reduce_xor: vector_reduce_xor, false);
2545 if name == sym::simd_reduce_all {
let input =
if !true {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
args[0].immediate()
} else {
let bitwidth =
match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_all,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth,
in_len as _)
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_and(input);
Ok(if !true { r } else { bx.zext(r, bx.type_bool()) })
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_all,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};bitwise_red!(simd_reduce_all: vector_reduce_and, true);
2546 if name == sym::simd_reduce_any {
let input =
if !true {
if !(ret_ty == in_elem) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnType {
span,
name,
in_elem,
in_ty,
ret_ty,
});
return Err(());
};
};
args[0].immediate()
} else {
let bitwidth =
match in_elem.kind() {
ty::Int(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
ty::Uint(i) => {
i.bit_width().unwrap_or_else(||
bx.data_layout().pointer_size().bits())
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_any,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
vector_mask_to_bitmask(bx, args[0].immediate(), bitwidth,
in_len as _)
};
return match in_elem.kind() {
ty::Int(_) | ty::Uint(_) => {
let r = bx.vector_reduce_or(input);
Ok(if !true { r } else { bx.zext(r, bx.type_bool()) })
}
_ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedSymbol {
span,
name,
symbol: sym::simd_reduce_any,
in_ty,
in_elem,
ret_ty,
});
return Err(());
}
};
};bitwise_red!(simd_reduce_any: vector_reduce_or, true);
2547
2548 if name == sym::simd_cast_ptr {
2549 let (out_len, out_elem) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
2550 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLengthInputType {
span,
name,
in_len,
in_ty,
ret_ty,
out_len,
});
return Err(());
};
};require!(
2551 in_len == out_len,
2552 InvalidMonomorphization::ReturnLengthInputType {
2553 span,
2554 name,
2555 in_len,
2556 in_ty,
2557 ret_ty,
2558 out_len
2559 }
2560 );
2561
2562 match in_elem.kind() {
2563 ty::RawPtr(p_ty, _) => {
2564 let metadata = p_ty.ptr_metadata_ty(bx.tcx, |ty| {
2565 bx.tcx.normalize_erasing_regions(bx.typing_env(), ty)
2566 });
2567 if !metadata.is_unit() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::CastWidePointer {
span,
name,
ty: in_elem,
});
return Err(());
};
};require!(
2568 metadata.is_unit(),
2569 InvalidMonomorphization::CastWidePointer { span, name, ty: in_elem }
2570 );
2571 }
2572 _ => {
2573 {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedPointer {
span,
name,
ty: in_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedPointer { span, name, ty: in_elem })
2574 }
2575 }
2576 match out_elem.kind() {
2577 ty::RawPtr(p_ty, _) => {
2578 let metadata = p_ty.ptr_metadata_ty(bx.tcx, |ty| {
2579 bx.tcx.normalize_erasing_regions(bx.typing_env(), ty)
2580 });
2581 if !metadata.is_unit() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::CastWidePointer {
span,
name,
ty: out_elem,
});
return Err(());
};
};require!(
2582 metadata.is_unit(),
2583 InvalidMonomorphization::CastWidePointer { span, name, ty: out_elem }
2584 );
2585 }
2586 _ => {
2587 {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedPointer {
span,
name,
ty: out_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedPointer { span, name, ty: out_elem })
2588 }
2589 }
2590
2591 return Ok(args[0].immediate());
2592 }
2593
2594 if name == sym::simd_expose_provenance {
2595 let (out_len, out_elem) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
2596 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLengthInputType {
span,
name,
in_len,
in_ty,
ret_ty,
out_len,
});
return Err(());
};
};require!(
2597 in_len == out_len,
2598 InvalidMonomorphization::ReturnLengthInputType {
2599 span,
2600 name,
2601 in_len,
2602 in_ty,
2603 ret_ty,
2604 out_len
2605 }
2606 );
2607
2608 match in_elem.kind() {
2609 ty::RawPtr(_, _) => {}
2610 _ => {
2611 {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedPointer {
span,
name,
ty: in_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedPointer { span, name, ty: in_elem })
2612 }
2613 }
2614 match out_elem.kind() {
2615 ty::Uint(ty::UintTy::Usize) => {}
2616 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedUsize {
span,
name,
ty: out_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedUsize { span, name, ty: out_elem }),
2617 }
2618
2619 return Ok(bx.ptrtoint(args[0].immediate(), llret_ty));
2620 }
2621
2622 if name == sym::simd_with_exposed_provenance {
2623 let (out_len, out_elem) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
2624 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLengthInputType {
span,
name,
in_len,
in_ty,
ret_ty,
out_len,
});
return Err(());
};
};require!(
2625 in_len == out_len,
2626 InvalidMonomorphization::ReturnLengthInputType {
2627 span,
2628 name,
2629 in_len,
2630 in_ty,
2631 ret_ty,
2632 out_len
2633 }
2634 );
2635
2636 match in_elem.kind() {
2637 ty::Uint(ty::UintTy::Usize) => {}
2638 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedUsize {
span,
name,
ty: in_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedUsize { span, name, ty: in_elem }),
2639 }
2640 match out_elem.kind() {
2641 ty::RawPtr(_, _) => {}
2642 _ => {
2643 {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedPointer {
span,
name,
ty: out_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::ExpectedPointer { span, name, ty: out_elem })
2644 }
2645 }
2646
2647 return Ok(bx.inttoptr(args[0].immediate(), llret_ty));
2648 }
2649
2650 if name == sym::simd_cast || name == sym::simd_as {
2651 let (out_len, out_elem) = {
if !ret_ty.is_simd() {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::SimdReturn {
span,
name,
ty: ret_ty,
});
return Err(());
};
};
ret_ty.simd_size_and_type(bx.tcx())
}require_simd!(ret_ty, SimdReturn);
2652 if !(in_len == out_len) {
{
bx.sess().dcx().emit_err(InvalidMonomorphization::ReturnLengthInputType {
span,
name,
in_len,
in_ty,
ret_ty,
out_len,
});
return Err(());
};
};require!(
2653 in_len == out_len,
2654 InvalidMonomorphization::ReturnLengthInputType {
2655 span,
2656 name,
2657 in_len,
2658 in_ty,
2659 ret_ty,
2660 out_len
2661 }
2662 );
2663 if in_elem == out_elem {
2665 return Ok(args[0].immediate());
2666 }
2667
2668 #[derive(#[automatically_derived]
impl ::core::marker::Copy for Sign { }Copy, #[automatically_derived]
impl ::core::clone::Clone for Sign {
#[inline]
fn clone(&self) -> Sign { *self }
}Clone)]
2669 enum Sign {
2670 Unsigned,
2671 Signed,
2672 }
2673 use Sign::*;
2674
2675 enum Style {
2676 Float,
2677 Int(Sign),
2678 Unsupported,
2679 }
2680
2681 let (in_style, in_width) = match in_elem.kind() {
2682 ty::Int(i) => (
2685 Style::Int(Signed),
2686 i.normalize(bx.tcx().sess.target.pointer_width).bit_width().unwrap(),
2687 ),
2688 ty::Uint(u) => (
2689 Style::Int(Unsigned),
2690 u.normalize(bx.tcx().sess.target.pointer_width).bit_width().unwrap(),
2691 ),
2692 ty::Float(f) => (Style::Float, f.bit_width()),
2693 _ => (Style::Unsupported, 0),
2694 };
2695 let (out_style, out_width) = match out_elem.kind() {
2696 ty::Int(i) => (
2697 Style::Int(Signed),
2698 i.normalize(bx.tcx().sess.target.pointer_width).bit_width().unwrap(),
2699 ),
2700 ty::Uint(u) => (
2701 Style::Int(Unsigned),
2702 u.normalize(bx.tcx().sess.target.pointer_width).bit_width().unwrap(),
2703 ),
2704 ty::Float(f) => (Style::Float, f.bit_width()),
2705 _ => (Style::Unsupported, 0),
2706 };
2707
2708 match (in_style, out_style) {
2709 (Style::Int(sign), Style::Int(_)) => {
2710 return Ok(match in_width.cmp(&out_width) {
2711 Ordering::Greater => bx.trunc(args[0].immediate(), llret_ty),
2712 Ordering::Equal => args[0].immediate(),
2713 Ordering::Less => match sign {
2714 Sign::Signed => bx.sext(args[0].immediate(), llret_ty),
2715 Sign::Unsigned => bx.zext(args[0].immediate(), llret_ty),
2716 },
2717 });
2718 }
2719 (Style::Int(Sign::Signed), Style::Float) => {
2720 return Ok(bx.sitofp(args[0].immediate(), llret_ty));
2721 }
2722 (Style::Int(Sign::Unsigned), Style::Float) => {
2723 return Ok(bx.uitofp(args[0].immediate(), llret_ty));
2724 }
2725 (Style::Float, Style::Int(sign)) => {
2726 return Ok(match (sign, name == sym::simd_as) {
2727 (Sign::Unsigned, false) => bx.fptoui(args[0].immediate(), llret_ty),
2728 (Sign::Signed, false) => bx.fptosi(args[0].immediate(), llret_ty),
2729 (_, true) => bx.cast_float_to_int(
2730 #[allow(non_exhaustive_omitted_patterns)] match sign {
Sign::Signed => true,
_ => false,
}matches!(sign, Sign::Signed),
2731 args[0].immediate(),
2732 llret_ty,
2733 ),
2734 });
2735 }
2736 (Style::Float, Style::Float) => {
2737 return Ok(match in_width.cmp(&out_width) {
2738 Ordering::Greater => bx.fptrunc(args[0].immediate(), llret_ty),
2739 Ordering::Equal => args[0].immediate(),
2740 Ordering::Less => bx.fpext(args[0].immediate(), llret_ty),
2741 });
2742 }
2743 _ => { }
2744 }
2745 {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedCast {
span,
name,
in_ty,
in_elem,
ret_ty,
out_elem,
});
return Err(());
};return_error!(InvalidMonomorphization::UnsupportedCast {
2746 span,
2747 name,
2748 in_ty,
2749 in_elem,
2750 ret_ty,
2751 out_elem
2752 });
2753 }
2754 macro_rules! arith_binary {
2755 ($($name: ident: $($($p: ident),* => $call: ident),*;)*) => {
2756 $(if name == sym::$name {
2757 match in_elem.kind() {
2758 $($(ty::$p(_))|* => {
2759 return Ok(bx.$call(args[0].immediate(), args[1].immediate()))
2760 })*
2761 _ => {},
2762 }
2763 return_error!(
2764 InvalidMonomorphization::UnsupportedOperation { span, name, in_ty, in_elem }
2765 );
2766 })*
2767 }
2768 }
2769 if name == sym::simd_fmin {
match in_elem.kind() {
ty::Float(_) => {
return Ok(bx.minnum(args[0].immediate(), args[1].immediate()))
}
_ => {}
}
{
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedOperation {
span,
name,
in_ty,
in_elem,
});
return Err(());
};
}arith_binary! {
2770 simd_add: Uint, Int => add, Float => fadd;
2771 simd_sub: Uint, Int => sub, Float => fsub;
2772 simd_mul: Uint, Int => mul, Float => fmul;
2773 simd_div: Uint => udiv, Int => sdiv, Float => fdiv;
2774 simd_rem: Uint => urem, Int => srem, Float => frem;
2775 simd_shl: Uint, Int => shl;
2776 simd_shr: Uint => lshr, Int => ashr;
2777 simd_and: Uint, Int => and;
2778 simd_or: Uint, Int => or;
2779 simd_xor: Uint, Int => xor;
2780 simd_fmax: Float => maxnum;
2781 simd_fmin: Float => minnum;
2782
2783 }
2784 macro_rules! arith_unary {
2785 ($($name: ident: $($($p: ident),* => $call: ident),*;)*) => {
2786 $(if name == sym::$name {
2787 match in_elem.kind() {
2788 $($(ty::$p(_))|* => {
2789 return Ok(bx.$call(args[0].immediate()))
2790 })*
2791 _ => {},
2792 }
2793 return_error!(
2794 InvalidMonomorphization::UnsupportedOperation { span, name, in_ty, in_elem }
2795 );
2796 })*
2797 }
2798 }
2799 if name == sym::simd_neg {
match in_elem.kind() {
ty::Int(_) => { return Ok(bx.neg(args[0].immediate())) }
ty::Float(_) => { return Ok(bx.fneg(args[0].immediate())) }
_ => {}
}
{
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedOperation {
span,
name,
in_ty,
in_elem,
});
return Err(());
};
}arith_unary! {
2800 simd_neg: Int => neg, Float => fneg;
2801 }
2802
2803 if #[allow(non_exhaustive_omitted_patterns)] match name {
sym::simd_bswap | sym::simd_bitreverse | sym::simd_ctlz | sym::simd_ctpop
| sym::simd_cttz | sym::simd_carryless_mul | sym::simd_funnel_shl |
sym::simd_funnel_shr => true,
_ => false,
}matches!(
2805 name,
2806 sym::simd_bswap
2807 | sym::simd_bitreverse
2808 | sym::simd_ctlz
2809 | sym::simd_ctpop
2810 | sym::simd_cttz
2811 | sym::simd_carryless_mul
2812 | sym::simd_funnel_shl
2813 | sym::simd_funnel_shr
2814 ) {
2815 let vec_ty = bx.cx.type_vector(
2816 match *in_elem.kind() {
2817 ty::Int(i) => bx.cx.type_int_from_ty(i),
2818 ty::Uint(i) => bx.cx.type_uint_from_ty(i),
2819 _ => {
bx.sess().dcx().emit_err(InvalidMonomorphization::UnsupportedOperation {
span,
name,
in_ty,
in_elem,
});
return Err(());
}return_error!(InvalidMonomorphization::UnsupportedOperation {
2820 span,
2821 name,
2822 in_ty,
2823 in_elem
2824 }),
2825 },
2826 in_len as u64,
2827 );
2828 let llvm_intrinsic = match name {
2829 sym::simd_bswap => "llvm.bswap",
2830 sym::simd_bitreverse => "llvm.bitreverse",
2831 sym::simd_ctlz => "llvm.ctlz",
2832 sym::simd_ctpop => "llvm.ctpop",
2833 sym::simd_cttz => "llvm.cttz",
2834 sym::simd_funnel_shl => "llvm.fshl",
2835 sym::simd_funnel_shr => "llvm.fshr",
2836 sym::simd_carryless_mul => "llvm.clmul",
2837 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2838 };
2839 let int_size = in_elem.int_size_and_signed(bx.tcx()).0.bits();
2840
2841 return match name {
2842 sym::simd_bswap if int_size == 8 => Ok(args[0].immediate()),
2844 sym::simd_ctlz | sym::simd_cttz => {
2845 let dont_poison_on_zero = bx.const_int(bx.type_i1(), 0);
2847 Ok(bx.call_intrinsic(
2848 llvm_intrinsic,
2849 &[vec_ty],
2850 &[args[0].immediate(), dont_poison_on_zero],
2851 ))
2852 }
2853 sym::simd_bswap | sym::simd_bitreverse | sym::simd_ctpop => {
2854 Ok(bx.call_intrinsic(llvm_intrinsic, &[vec_ty], &[args[0].immediate()]))
2856 }
2857 sym::simd_funnel_shl | sym::simd_funnel_shr => Ok(bx.call_intrinsic(
2858 llvm_intrinsic,
2859 &[vec_ty],
2860 &[args[0].immediate(), args[1].immediate(), args[2].immediate()],
2861 )),
2862 sym::simd_carryless_mul => {
2863 if crate::llvm_util::get_version() >= (22, 0, 0) {
2864 Ok(bx.call_intrinsic(
2865 llvm_intrinsic,
2866 &[vec_ty],
2867 &[args[0].immediate(), args[1].immediate()],
2868 ))
2869 } else {
2870 ::rustc_middle::util::bug::span_bug_fmt(span,
format_args!("`simd_carryless_mul` needs LLVM 22 or higher"));span_bug!(span, "`simd_carryless_mul` needs LLVM 22 or higher");
2871 }
2872 }
2873 _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2874 };
2875 }
2876
2877 if name == sym::simd_arith_offset {
2878 let pointee = in_elem.builtin_deref(true).unwrap_or_else(|| {
2880 ::rustc_middle::util::bug::span_bug_fmt(span,
format_args!("must be called with a vector of pointer types as first argument"))span_bug!(span, "must be called with a vector of pointer types as first argument")
2881 });
2882 let layout = bx.layout_of(pointee);
2883 let ptrs = args[0].immediate();
2884 let (_offsets_len, offsets_elem) = args[1].layout.ty.simd_size_and_type(bx.tcx());
2887 if !#[allow(non_exhaustive_omitted_patterns)] match offsets_elem.kind() {
ty::Int(ty::IntTy::Isize) | ty::Uint(ty::UintTy::Usize) => true,
_ => false,
}matches!(offsets_elem.kind(), ty::Int(ty::IntTy::Isize) | ty::Uint(ty::UintTy::Usize)) {
2888 ::rustc_middle::util::bug::span_bug_fmt(span,
format_args!("must be called with a vector of pointer-sized integers as second argument"));span_bug!(
2889 span,
2890 "must be called with a vector of pointer-sized integers as second argument"
2891 );
2892 }
2893 let offsets = args[1].immediate();
2894
2895 return Ok(bx.gep(bx.backend_type(layout), ptrs, &[offsets]));
2896 }
2897
2898 if name == sym::simd_saturating_add || name == sym::simd_saturating_sub {
2899 let lhs = args[0].immediate();
2900 let rhs = args[1].immediate();
2901 let is_add = name == sym::simd_saturating_add;
2902 let (signed, elem_ty) = match *in_elem.kind() {
2903 ty::Int(i) => (true, bx.cx.type_int_from_ty(i)),
2904 ty::Uint(i) => (false, bx.cx.type_uint_from_ty(i)),
2905 _ => {
2906 {
bx.sess().dcx().emit_err(InvalidMonomorphization::ExpectedVectorElementType {
span,
name,
expected_element: args[0].layout.ty.simd_size_and_type(bx.tcx()).1,
vector_type: args[0].layout.ty,
});
return Err(());
};return_error!(InvalidMonomorphization::ExpectedVectorElementType {
2907 span,
2908 name,
2909 expected_element: args[0].layout.ty.simd_size_and_type(bx.tcx()).1,
2910 vector_type: args[0].layout.ty
2911 });
2912 }
2913 };
2914 let llvm_intrinsic = ::alloc::__export::must_use({
::alloc::fmt::format(format_args!("llvm.{0}{1}.sat",
if signed { 's' } else { 'u' },
if is_add { "add" } else { "sub" }))
})format!(
2915 "llvm.{}{}.sat",
2916 if signed { 's' } else { 'u' },
2917 if is_add { "add" } else { "sub" },
2918 );
2919 let vec_ty = bx.cx.type_vector(elem_ty, in_len as u64);
2920
2921 return Ok(bx.call_intrinsic(llvm_intrinsic, &[vec_ty], &[lhs, rhs]));
2922 }
2923
2924 ::rustc_middle::util::bug::span_bug_fmt(span,
format_args!("unknown SIMD intrinsic"));span_bug!(span, "unknown SIMD intrinsic");
2925}