src/librustc_target/abi/call/nvptx64.rs

   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::{ArgType, FnType};
   5
   6 fn classify_ret_ty<Ty>(ret: &mut ArgType<'_, 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
  14 fn classify_arg_ty<Ty>(arg: &mut ArgType<'_, 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
  22 pub fn compute_abi_info<Ty>(fty: &mut FnType<'_, Ty>) {
  23     if !fty.ret.is_ignore() {
  24         classify_ret_ty(&mut fty.ret);
  25     }
  26
  27     for arg in &mut fty.args {
  28         if arg.is_ignore() {
  29             continue;
  30         }
  31         classify_arg_ty(arg);
  32     }
  33 }