[flang-commits] [flang] [flang][cuda] Add interfaces for __float2int_rX and __float2unit_rX (PR #153691)
Valentin Clement バレンタイン クレメン via flang-commits
flang-commits at lists.llvm.org
Thu Aug 14 15:27:48 PDT 2025
https://github.com/clementval updated https://github.com/llvm/llvm-project/pull/153691
>From 6daedf9935cb1d2b95aa4f89c0af28905ec6002e Mon Sep 17 00:00:00 2001
From: Valentin Clement <clementval at gmail.com>
Date: Thu, 14 Aug 2025 14:28:35 -0700
Subject: [PATCH] [flang][cuda] Add interfaces for __float2int_rX and
__float2unit_rX
---
flang/module/cudadevice.f90 | 56 ++++++++++++++++++++++++
flang/test/Lower/CUDA/cuda-libdevice.cuf | 30 +++++++++++++
2 files changed, 86 insertions(+)
diff --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90
index 1e403b1ffdb91..cc84662466116 100644
--- a/flang/module/cudadevice.f90
+++ b/flang/module/cudadevice.f90
@@ -471,6 +471,62 @@ attributes(device) real(8) function sinpi(x) bind(c,name='__nv_sinpi')
end function
end interface
+ interface __float2int_rd
+ attributes(device) integer function __float2int_rd(r) bind(c, name='__nv_float2int_rd')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2int_rn
+ attributes(device) integer function __float2int_rn(r) bind(c, name='__nv_float2int_rn')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2int_ru
+ attributes(device) integer function __float2int_ru(r) bind(c, name='__nv_float2int_ru')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2int_rz
+ attributes(device) integer function __float2int_rz(r) bind(c, name='__nv_float2int_rz')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2uint_rd
+ attributes(device) integer function __float2uint_rd(r) bind(c, name='__nv_float2uint_rd')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2uint_rn
+ attributes(device) integer function __float2uint_rn(r) bind(c, name='__nv_float2uint_rn')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2uint_ru
+ attributes(device) integer function __float2uint_ru(r) bind(c, name='__nv_float2uint_ru')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
+ interface __float2uint_rz
+ attributes(device) integer function __float2uint_rz(r) bind(c, name='__nv_float2uint_rz')
+ !dir$ ignore_tkr (d) r
+ real, value :: r
+ end function
+ end interface
+
interface __half2float
attributes(device) real function __half2float(i) bind(c, name='__nv_half2float')
!dir$ ignore_tkr (d) i
diff --git a/flang/test/Lower/CUDA/cuda-libdevice.cuf b/flang/test/Lower/CUDA/cuda-libdevice.cuf
index 844bdb954924a..8e8f1c8e4ec76 100644
--- a/flang/test/Lower/CUDA/cuda-libdevice.cuf
+++ b/flang/test/Lower/CUDA/cuda-libdevice.cuf
@@ -115,3 +115,33 @@ end subroutine
! CHECK: %{{.*}} = fir.call @__nv_double2ll_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
! CHECK: %{{.*}} = fir.call @__nv_double2ll_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
! CHECK: %{{.*}} = fir.call @__nv_double2ll_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
+
+attributes(global) subroutine test_float2int_rX()
+ integer :: res
+ real :: r
+ res = __float2int_rd(r)
+ res = __float2int_rn(r)
+ res = __float2int_ru(r)
+ res = __float2int_rz(r)
+end subroutine
+
+! CHECK-LABEL: _QPtest_float2int_rx
+! CHECK: %{{.*}} = fir.call @__nv_float2int_rd(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2int_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2int_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2int_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+
+attributes(global) subroutine test_float2uint_rX()
+ integer :: res
+ real :: r
+ res = __float2uint_rd(r)
+ res = __float2uint_rn(r)
+ res = __float2uint_ru(r)
+ res = __float2uint_rz(r)
+end subroutine
+
+! CHECK-LABEL: _QPtest_float2uint_rx
+! CHECK: %{{.*}} = fir.call @__nv_float2uint_rd(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2uint_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2uint_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
+! CHECK: %{{.*}} = fir.call @__nv_float2uint_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
More information about the flang-commits
mailing list