Skip to content

Commit 02e3618

Browse files
committed
Run cargo fmt
1 parent 0002297 commit 02e3618

File tree

16 files changed

+325
-244
lines changed

16 files changed

+325
-244
lines changed

crates/cuda_std/src/lib.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
#![allow(internal_features)]
2525
#![cfg_attr(
2626
target_os = "cuda",
27-
feature(alloc_error_handler, asm_experimental_arch, link_llvm_intrinsics),
27+
feature(alloc_error_handler, asm_experimental_arch, link_llvm_intrinsics)
2828
)]
2929

3030
extern crate alloc;

crates/optix/examples/ex02_pipeline/device/src/lib.rs

+1-5
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,5 @@
11
#![feature(asm)]
2-
#![cfg_attr(
3-
target_os = "cuda",
4-
no_std,
5-
register_attr(nvvm_internal)
6-
)]
2+
#![cfg_attr(target_os = "cuda", no_std, register_attr(nvvm_internal))]
73
// #![deny(warnings)]
84
#![allow(clippy::missing_safety_doc)]
95

crates/rustc_codegen_nvvm/src/abi.rs

+22-8
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,16 @@ impl LlvmType for CastTarget {
207207
self.rest.total
208208
);
209209
if self.rest.total.bytes() % self.rest.unit.size.bytes() != 0 {
210-
assert_eq!(self.rest.unit.kind, RegKind::Integer, "only int regs can be split");
210+
assert_eq!(
211+
self.rest.unit.kind,
212+
RegKind::Integer,
213+
"only int regs can be split"
214+
);
211215
}
212-
self.rest.total.bytes().div_ceil(self.rest.unit.size.bytes())
216+
self.rest
217+
.total
218+
.bytes()
219+
.div_ceil(self.rest.unit.size.bytes())
213220
};
214221

215222
// Simplify to a single unit or an array if there's no prefix.
@@ -226,8 +233,10 @@ impl LlvmType for CastTarget {
226233
}
227234

228235
// Generate a struct type with the prefix and the "rest" arguments.
229-
let prefix_args =
230-
self.prefix.iter().flat_map(|option_reg| option_reg.map(|reg| reg.llvm_type(cx)));
236+
let prefix_args = self
237+
.prefix
238+
.iter()
239+
.flat_map(|option_reg| option_reg.map(|reg| reg.llvm_type(cx)));
231240
let rest_args = (0..rest_count).map(|_| rest_ll_unit);
232241
let args: Vec<_> = prefix_args.chain(rest_args).collect();
233242
cx.type_struct(&args, false)
@@ -597,8 +606,7 @@ impl<'ll, 'tcx> ArgAbiExt<'ll, 'tcx> for ArgAbi<'tcx, Ty<'tcx>> {
597606
on_stack: _,
598607
} => {
599608
let align = attrs.pointee_align.unwrap_or(self.layout.align.abi);
600-
OperandValue::Ref(PlaceValue::new_sized(val, align))
601-
.store(bx, dst);
609+
OperandValue::Ref(PlaceValue::new_sized(val, align)).store(bx, dst);
602610
}
603611
// Unsized indirect arguments
604612
PassMode::Indirect {
@@ -642,7 +650,9 @@ impl<'ll, 'tcx> ArgAbiExt<'ll, 'tcx> for ArgAbi<'tcx, Ty<'tcx>> {
642650
trace!("store cast end");
643651
}
644652
_ => {
645-
OperandRef::from_immediate_or_packed_pair(bx, val, self.layout).val.store(bx, dst);
653+
OperandRef::from_immediate_or_packed_pair(bx, val, self.layout)
654+
.val
655+
.store(bx, dst);
646656
}
647657
}
648658
}
@@ -676,7 +686,11 @@ impl<'ll, 'tcx> ArgAbiExt<'ll, 'tcx> for ArgAbi<'tcx, Ty<'tcx>> {
676686
OperandValue::Ref(place_val).store(bx, dst);
677687
}
678688
PassMode::Direct(_)
679-
| PassMode::Indirect { attrs: _, meta_attrs: None, on_stack: _, }
689+
| PassMode::Indirect {
690+
attrs: _,
691+
meta_attrs: None,
692+
on_stack: _,
693+
}
680694
| PassMode::Cast { .. } => {
681695
let next_arg = next();
682696
self.store(bx, next_arg, dst);

crates/rustc_codegen_nvvm/src/allocator.rs

+20-14
Original file line numberDiff line numberDiff line change
@@ -48,12 +48,14 @@ pub(crate) unsafe fn codegen(
4848
}
4949
};
5050

51-
let ty = unsafe { llvm::LLVMFunctionType(
52-
output.unwrap_or(void),
53-
args.as_ptr(),
54-
args.len() as c_uint,
55-
False,
56-
) };
51+
let ty = unsafe {
52+
llvm::LLVMFunctionType(
53+
output.unwrap_or(void),
54+
args.as_ptr(),
55+
args.len() as c_uint,
56+
False,
57+
)
58+
};
5759
let name = format!("__rust_{}", method.name);
5860
let llfn = unsafe {
5961
llvm::LLVMRustGetOrInsertFunction(llmod, name.as_ptr().cast(), name.len(), ty)
@@ -68,7 +70,9 @@ pub(crate) unsafe fn codegen(
6870
};
6971
unsafe { llvm::LLVMRustSetVisibility(callee, llvm::Visibility::Hidden) };
7072

71-
let llbb = unsafe { llvm::LLVMAppendBasicBlockInContext(llcx, llfn, "entry\0".as_ptr().cast()) };
73+
let llbb = unsafe {
74+
llvm::LLVMAppendBasicBlockInContext(llcx, llfn, "entry\0".as_ptr().cast())
75+
};
7276

7377
let llbuilder = unsafe { llvm::LLVMCreateBuilderInContext(llcx) };
7478
unsafe { llvm::LLVMPositionBuilderAtEnd(llbuilder, llbb) };
@@ -77,13 +81,15 @@ pub(crate) unsafe fn codegen(
7781
.enumerate()
7882
.map(|(i, _)| unsafe { llvm::LLVMGetParam(llfn, i as c_uint) })
7983
.collect::<Vec<_>>();
80-
let ret = unsafe { llvm::LLVMRustBuildCall(
81-
llbuilder,
82-
callee,
83-
args.as_ptr(),
84-
args.len() as c_uint,
85-
None,
86-
) };
84+
let ret = unsafe {
85+
llvm::LLVMRustBuildCall(
86+
llbuilder,
87+
callee,
88+
args.as_ptr(),
89+
args.len() as c_uint,
90+
None,
91+
)
92+
};
8793
unsafe { llvm::LLVMSetTailCall(ret, True) };
8894
if output.is_some() {
8995
unsafe { llvm::LLVMBuildRet(llbuilder, ret) };

crates/rustc_codegen_nvvm/src/back.rs

+103-97
Original file line numberDiff line numberDiff line change
@@ -194,8 +194,9 @@ pub(crate) unsafe fn codegen(
194194
.temp_path(OutputType::LlvmAssembly, module_name);
195195
let out = out.to_str().unwrap();
196196

197-
let result =
198-
unsafe { llvm::LLVMRustPrintModule(llmod, out.as_c_char_ptr(), out.len(), demangle_callback) };
197+
let result = unsafe {
198+
llvm::LLVMRustPrintModule(llmod, out.as_c_char_ptr(), out.len(), demangle_callback)
199+
};
199200

200201
result.into_result().map_err(|()| {
201202
let msg = format!("failed to write NVVM IR to {}", out);
@@ -350,56 +351,59 @@ pub(crate) unsafe fn optimize(
350351

351352
let tm = (cgcx.tm_factory)(tm_factory_config).expect("failed to create target machine");
352353

353-
if config.opt_level.is_some() { unsafe {
354-
let fpm = llvm::LLVMCreateFunctionPassManagerForModule(llmod);
355-
let mpm = llvm::LLVMCreatePassManager();
354+
if config.opt_level.is_some() {
355+
unsafe {
356+
let fpm = llvm::LLVMCreateFunctionPassManagerForModule(llmod);
357+
let mpm = llvm::LLVMCreatePassManager();
356358

357-
let addpass = |pass_name: &str| {
358-
let pass = llvm::LLVMRustFindAndCreatePass(pass_name.as_c_char_ptr(), pass_name.len());
359-
if pass.is_none() {
360-
return false;
361-
}
362-
let pass = pass.unwrap();
363-
let pass_manager = match llvm::LLVMRustPassKind(pass) {
364-
llvm::PassKind::Function => &fpm,
365-
llvm::PassKind::Module => &mpm,
366-
llvm::PassKind::Other => {
367-
diag_handler.err("Encountered LLVM pass kind we can't handle");
368-
return true;
359+
let addpass = |pass_name: &str| {
360+
let pass =
361+
llvm::LLVMRustFindAndCreatePass(pass_name.as_c_char_ptr(), pass_name.len());
362+
if pass.is_none() {
363+
return false;
369364
}
365+
let pass = pass.unwrap();
366+
let pass_manager = match llvm::LLVMRustPassKind(pass) {
367+
llvm::PassKind::Function => &fpm,
368+
llvm::PassKind::Module => &mpm,
369+
llvm::PassKind::Other => {
370+
diag_handler.err("Encountered LLVM pass kind we can't handle");
371+
return true;
372+
}
373+
};
374+
llvm::LLVMRustAddPass(pass_manager, pass);
375+
true
370376
};
371-
llvm::LLVMRustAddPass(pass_manager, pass);
372-
true
373-
};
374377

375-
if !config.no_prepopulate_passes {
376-
llvm::LLVMRustAddAnalysisPasses(tm, fpm, llmod);
377-
llvm::LLVMRustAddAnalysisPasses(tm, mpm, llmod);
378-
let opt_level = config
379-
.opt_level
380-
.map_or(llvm::CodeGenOptLevel::None, |x| to_llvm_opt_settings(x).0);
381-
with_llvm_pmb(llmod, config, opt_level, &mut |b| {
382-
llvm::LLVMPassManagerBuilderPopulateFunctionPassManager(b, fpm);
383-
llvm::LLVMPassManagerBuilderPopulateModulePassManager(b, mpm);
384-
})
385-
}
378+
if !config.no_prepopulate_passes {
379+
llvm::LLVMRustAddAnalysisPasses(tm, fpm, llmod);
380+
llvm::LLVMRustAddAnalysisPasses(tm, mpm, llmod);
381+
let opt_level = config
382+
.opt_level
383+
.map_or(llvm::CodeGenOptLevel::None, |x| to_llvm_opt_settings(x).0);
384+
with_llvm_pmb(llmod, config, opt_level, &mut |b| {
385+
llvm::LLVMPassManagerBuilderPopulateFunctionPassManager(b, fpm);
386+
llvm::LLVMPassManagerBuilderPopulateModulePassManager(b, mpm);
387+
})
388+
}
386389

387-
for pass in &config.passes {
388-
if !addpass(pass) {
389-
diag_handler.warn(format!("unknown pass `{}`, ignoring", pass));
390+
for pass in &config.passes {
391+
if !addpass(pass) {
392+
diag_handler.warn(format!("unknown pass `{}`, ignoring", pass));
393+
}
390394
}
391-
}
392395

393-
diag_handler.abort_if_errors();
396+
diag_handler.abort_if_errors();
394397

395-
// Finally, run the actual optimization passes
396-
llvm::LLVMRustRunFunctionPassManager(fpm, llmod);
397-
llvm::LLVMRunPassManager(mpm, llmod);
398+
// Finally, run the actual optimization passes
399+
llvm::LLVMRustRunFunctionPassManager(fpm, llmod);
400+
llvm::LLVMRunPassManager(mpm, llmod);
398401

399-
// Deallocate managers that we're now done with
400-
llvm::LLVMDisposePassManager(fpm);
401-
llvm::LLVMDisposePassManager(mpm);
402-
} }
402+
// Deallocate managers that we're now done with
403+
llvm::LLVMDisposePassManager(fpm);
404+
llvm::LLVMDisposePassManager(mpm);
405+
}
406+
}
403407

404408
Ok(())
405409
}
@@ -409,61 +413,63 @@ unsafe fn with_llvm_pmb(
409413
config: &ModuleConfig,
410414
opt_level: llvm::CodeGenOptLevel,
411415
f: &mut impl FnMut(&llvm::PassManagerBuilder),
412-
) { unsafe {
413-
use std::ptr;
414-
415-
let builder = llvm::LLVMPassManagerBuilderCreate();
416-
let opt_size = config
417-
.opt_size
418-
.map_or(llvm::CodeGenOptSizeNone, |x| to_llvm_opt_settings(x).1);
419-
420-
llvm::LLVMRustConfigurePassManagerBuilder(
421-
builder,
422-
opt_level,
423-
config.merge_functions,
424-
config.vectorize_slp,
425-
config.vectorize_loop,
426-
false,
427-
ptr::null(),
428-
ptr::null(),
429-
);
430-
431-
llvm::LLVMPassManagerBuilderSetSizeLevel(builder, opt_size as u32);
432-
433-
if opt_size != llvm::CodeGenOptSizeNone {
434-
llvm::LLVMPassManagerBuilderSetDisableUnrollLoops(builder, 1);
435-
}
416+
) {
417+
unsafe {
418+
use std::ptr;
419+
420+
let builder = llvm::LLVMPassManagerBuilderCreate();
421+
let opt_size = config
422+
.opt_size
423+
.map_or(llvm::CodeGenOptSizeNone, |x| to_llvm_opt_settings(x).1);
424+
425+
llvm::LLVMRustConfigurePassManagerBuilder(
426+
builder,
427+
opt_level,
428+
config.merge_functions,
429+
config.vectorize_slp,
430+
config.vectorize_loop,
431+
false,
432+
ptr::null(),
433+
ptr::null(),
434+
);
435+
436+
llvm::LLVMPassManagerBuilderSetSizeLevel(builder, opt_size as u32);
437+
438+
if opt_size != llvm::CodeGenOptSizeNone {
439+
llvm::LLVMPassManagerBuilderSetDisableUnrollLoops(builder, 1);
440+
}
436441

437-
llvm::LLVMRustAddBuilderLibraryInfo(builder, llmod, config.no_builtins);
442+
llvm::LLVMRustAddBuilderLibraryInfo(builder, llmod, config.no_builtins);
438443

439-
// Here we match what clang does (kinda). For O0 we only inline
440-
// always-inline functions (but don't add lifetime intrinsics), at O1 we
441-
// inline with lifetime intrinsics, and O2+ we add an inliner with a
442-
// thresholds copied from clang.
443-
match (opt_level, opt_size) {
444-
(llvm::CodeGenOptLevel::Aggressive, ..) => {
445-
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 275);
446-
}
447-
(_, llvm::CodeGenOptSizeDefault) => {
448-
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 75);
449-
}
450-
(_, llvm::CodeGenOptSizeAggressive) => {
451-
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 25);
452-
}
453-
(llvm::CodeGenOptLevel::None, ..) => {
454-
llvm::LLVMRustAddAlwaysInlinePass(builder, false);
455-
}
456-
(llvm::CodeGenOptLevel::Less, ..) => {
457-
llvm::LLVMRustAddAlwaysInlinePass(builder, true);
458-
}
459-
(llvm::CodeGenOptLevel::Default, ..) => {
460-
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 225);
461-
}
462-
(llvm::CodeGenOptLevel::Other, ..) => {
463-
bug!("CodeGenOptLevel::Other selected")
444+
// Here we match what clang does (kinda). For O0 we only inline
445+
// always-inline functions (but don't add lifetime intrinsics), at O1 we
446+
// inline with lifetime intrinsics, and O2+ we add an inliner with a
447+
// thresholds copied from clang.
448+
match (opt_level, opt_size) {
449+
(llvm::CodeGenOptLevel::Aggressive, ..) => {
450+
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 275);
451+
}
452+
(_, llvm::CodeGenOptSizeDefault) => {
453+
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 75);
454+
}
455+
(_, llvm::CodeGenOptSizeAggressive) => {
456+
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 25);
457+
}
458+
(llvm::CodeGenOptLevel::None, ..) => {
459+
llvm::LLVMRustAddAlwaysInlinePass(builder, false);
460+
}
461+
(llvm::CodeGenOptLevel::Less, ..) => {
462+
llvm::LLVMRustAddAlwaysInlinePass(builder, true);
463+
}
464+
(llvm::CodeGenOptLevel::Default, ..) => {
465+
llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 225);
466+
}
467+
(llvm::CodeGenOptLevel::Other, ..) => {
468+
bug!("CodeGenOptLevel::Other selected")
469+
}
464470
}
465-
}
466471

467-
f(builder);
468-
llvm::LLVMPassManagerBuilderDispose(builder);
469-
}}
472+
f(builder);
473+
llvm::LLVMPassManagerBuilderDispose(builder);
474+
}
475+
}

0 commit comments

Comments
 (0)