44 lines
1.2 KiB
Rust
44 lines
1.2 KiB
Rust
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
// Reference: PTX Writer's Guide to Interoperability
|
|
// http://docs.nvidia.com/cuda/ptx-writers-guide-to-interoperability
|
|
|
|
use abi::{ArgType, FnType, LayoutExt};
|
|
|
|
fn classify_ret_ty(ret: &mut ArgType) {
|
|
if ret.layout.is_aggregate() && ret.layout.size.bits() > 64 {
|
|
ret.make_indirect();
|
|
} else {
|
|
ret.extend_integer_width_to(64);
|
|
}
|
|
}
|
|
|
|
fn classify_arg_ty(arg: &mut ArgType) {
|
|
if arg.layout.is_aggregate() && arg.layout.size.bits() > 64 {
|
|
arg.make_indirect();
|
|
} else {
|
|
arg.extend_integer_width_to(64);
|
|
}
|
|
}
|
|
|
|
pub fn compute_abi_info(fty: &mut FnType) {
|
|
if !fty.ret.is_ignore() {
|
|
classify_ret_ty(&mut fty.ret);
|
|
}
|
|
|
|
for arg in &mut fty.args {
|
|
if arg.is_ignore() {
|
|
continue;
|
|
}
|
|
classify_arg_ty(arg);
|
|
}
|
|
}
|