summaryrefslogtreecommitdiffstats
path: root/tests/assembly/nvptx-safe-naming.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/assembly/nvptx-safe-naming.rs')
-rw-r--r--tests/assembly/nvptx-safe-naming.rs38
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/assembly/nvptx-safe-naming.rs b/tests/assembly/nvptx-safe-naming.rs
new file mode 100644
index 000000000..80bb04fc0
--- /dev/null
+++ b/tests/assembly/nvptx-safe-naming.rs
@@ -0,0 +1,38 @@
+// assembly-output: ptx-linker
+// compile-flags: --crate-type cdylib
+// only-nvptx64
+// ignore-nvptx64
+
+#![feature(abi_ptx)]
+#![no_std]
+
+// aux-build: breakpoint-panic-handler.rs
+extern crate breakpoint_panic_handler;
+
+// Verify function name doesn't contain unacceaptable characters.
+// CHECK: .func (.param .b32 func_retval0) [[IMPL_FN:[a-zA-Z0-9$_]+square[a-zA-Z0-9$_]+]](
+
+// CHECK-LABEL: .visible .entry top_kernel(
+#[no_mangle]
+pub unsafe extern "ptx-kernel" fn top_kernel(a: *const u32, b: *mut u32) {
+ // CHECK: call.uni (retval0),
+ // CHECK-NEXT: [[IMPL_FN]]
+ *b = deep::private::MyStruct::new(*a).square();
+}
+
+pub mod deep {
+ pub mod private {
+ pub struct MyStruct<T>(T);
+
+ impl MyStruct<u32> {
+ pub fn new(a: u32) -> Self {
+ MyStruct(a)
+ }
+
+ #[inline(never)]
+ pub fn square(&self) -> u32 {
+ self.0.wrapping_mul(self.0)
+ }
+ }
+ }
+}