src/test/assembly/asm/nvptx-types.rs

   1 // min-llvm-version: 10.0.1
   2 // assembly-output: emit-asm
   3 // compile-flags: --target nvptx64-nvidia-cuda
   4 // compile-flags: --crate-type cdylib
   5 // needs-llvm-components: nvptx
   6
   7 #![feature(no_core, lang_items, rustc_attrs)]
   8 #![no_core]
   9
  10 #[rustc_builtin_macro]
  11 macro_rules! asm {
  12     () => {};
  13 }
  14 #[rustc_builtin_macro]
  15 macro_rules! concat {
  16     () => {};
  17 }
  18
  19 #[lang = "sized"]
  20 trait Sized {}
  21 #[lang = "copy"]
  22 trait Copy {}
  23
  24 type ptr = *mut u8;
  25
  26 impl Copy for i8 {}
  27 impl Copy for i16 {}
  28 impl Copy for i32 {}
  29 impl Copy for f32 {}
  30 impl Copy for i64 {}
  31 impl Copy for f64 {}
  32 impl Copy for ptr {}
  33
  34 // NVPTX does not support static variables
  35 #[no_mangle]
  36 fn extern_func() {}
  37
  38 // CHECK-LABEL: .visible .func sym_fn()
  39 // CHECK: // begin inline asm
  40 // CHECK: call extern_func;
  41 // CHECK: // end inline asm
  42 #[no_mangle]
  43 pub unsafe fn sym_fn() {
  44     asm!("call {};", sym extern_func);
  45 }
  46
  47 macro_rules! check {
  48     ($func:ident $ty:ident $class:ident $mov:literal) => {
  49         #[no_mangle]
  50         pub unsafe fn $func(x: $ty) -> $ty {
  51             let y;
  52             asm!(concat!($mov, " {}, {};"), out($class) y, in($class) x);
  53             y
  54         }
  55     };
  56 }
  57
  58 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg16_i8
  59 // CHECK: // begin inline asm
  60 // CHECK: mov.i16 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  61 // CHECK: // end inline asm
  62 check!(reg16_i8 i8 reg16 "mov.i16");
  63
  64 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg16_i16
  65 // CHECK: // begin inline asm
  66 // CHECK: mov.i16 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  67 // CHECK: // end inline asm
  68 check!(reg16_i16 i16 reg16 "mov.i16");
  69
  70 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg32_i8
  71 // CHECK: // begin inline asm
  72 // CHECK: mov.i32 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  73 // CHECK: // end inline asm
  74 check!(reg32_i8 i8 reg32 "mov.i32");
  75
  76 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg32_i16
  77 // CHECK: // begin inline asm
  78 // CHECK: mov.i32 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  79 // CHECK: // end inline asm
  80 check!(reg32_i16 i16 reg32 "mov.i32");
  81
  82 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg32_i32
  83 // CHECK: // begin inline asm
  84 // CHECK: mov.i32 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  85 // CHECK: // end inline asm
  86 check!(reg32_i32 i32 reg32 "mov.i32");
  87
  88 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg32_f32
  89 // CHECK: // begin inline asm
  90 // CHECK: mov.i32 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  91 // CHECK: // end inline asm
  92 check!(reg32_f32 f32 reg32 "mov.i32");
  93
  94 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg64_i8
  95 // CHECK: // begin inline asm
  96 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
  97 // CHECK: // end inline asm
  98 check!(reg64_i8 i8 reg64 "mov.i64");
  99
 100 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg64_i16
 101 // CHECK: // begin inline asm
 102 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 103 // CHECK: // end inline asm
 104 check!(reg64_i16 i16 reg64 "mov.i64");
 105
 106 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg64_i32
 107 // CHECK: // begin inline asm
 108 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 109 // CHECK: // end inline asm
 110 check!(reg64_i32 i32 reg64 "mov.i64");
 111
 112 // CHECK-LABEL: .visible .func (.param .b32 func_retval0) reg64_f32
 113 // CHECK: // begin inline asm
 114 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 115 // CHECK: // end inline asm
 116 check!(reg64_f32 f32 reg64 "mov.i64");
 117
 118 // CHECK-LABEL: .visible .func (.param .b64 func_retval0) reg64_i64
 119 // CHECK: // begin inline asm
 120 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 121 // CHECK: // end inline asm
 122 check!(reg64_i64 i64 reg64 "mov.i64");
 123
 124 // CHECK-LABEL: .visible .func (.param .b64 func_retval0) reg64_f64
 125 // CHECK: // begin inline asm
 126 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 127 // CHECK: // end inline asm
 128 check!(reg64_f64 f64 reg64 "mov.i64");
 129
 130 // CHECK-LABEL: .visible .func (.param .b64 func_retval0) reg64_ptr
 131 // CHECK: // begin inline asm
 132 // CHECK: mov.i64 %{{[a-z0-9]+}}, %{{[a-z0-9]+}};
 133 // CHECK: // end inline asm
 134 check!(reg64_ptr ptr reg64 "mov.i64");