1 // Reference: PTX Writer's Guide to Interoperability
2 // http://docs.nvidia.com/cuda/ptx-writers-guide-to-interoperability
3 
4 use crate::abi::call::{ArgAbi, FnAbi};
5 
classify_ret<Ty>(ret: &mut ArgAbi<'_, Ty>)6 fn classify_ret<Ty>(ret: &mut ArgAbi<'_, Ty>) {
7     if ret.layout.is_aggregate() && ret.layout.size.bits() > 64 {
8         ret.make_indirect();
9     } else {
10         ret.extend_integer_width_to(64);
11     }
12 }
13 
classify_arg<Ty>(arg: &mut ArgAbi<'_, Ty>)14 fn classify_arg<Ty>(arg: &mut ArgAbi<'_, Ty>) {
15     if arg.layout.is_aggregate() && arg.layout.size.bits() > 64 {
16         arg.make_indirect();
17     } else {
18         arg.extend_integer_width_to(64);
19     }
20 }
21 
compute_abi_info<Ty>(fn_abi: &mut FnAbi<'_, Ty>)22 pub fn compute_abi_info<Ty>(fn_abi: &mut FnAbi<'_, Ty>) {
23     if !fn_abi.ret.is_ignore() {
24         classify_ret(&mut fn_abi.ret);
25     }
26 
27     for arg in &mut fn_abi.args {
28         if arg.is_ignore() {
29             continue;
30         }
31         classify_arg(arg);
32     }
33 }
34