[flang-commits] [flang] [flang][cuda] Add interfaces for int_as_float and float_as_int (PR #153716)
Valentin Clement バレンタイン クレメン via flang-commits
flang-commits at lists.llvm.org
Fri Aug 15 10:00:53 PDT 2025
https://github.com/clementval updated https://github.com/llvm/llvm-project/pull/153716
>From 8be51fc8c21ce928bd14be5fcd74c66498f00def Mon Sep 17 00:00:00 2001
From: Valentin Clement <clementval at gmail.com>
Date: Thu, 14 Aug 2025 17:17:56 -0700
Subject: [PATCH] [flang][cuda] Add interfaces for int_as_float and
float_as_int
---
flang/module/cudadevice.f90 | 14 ++++++++++++++
flang/test/Lower/CUDA/cuda-libdevice.cuf | 18 ++++++++++++++++++
2 files changed, 32 insertions(+)
diff --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90
index ffc3a3b170ca6..9e7dfcf05a1ac 100644
--- a/flang/module/cudadevice.f90
+++ b/flang/module/cudadevice.f90
@@ -471,6 +471,20 @@ attributes(device) real(8) function sinpi(x) bind(c,name='__nv_sinpi')
end function
end interface
+ interface int_as_float
+ attributes(device) real function __int_as_float(i) bind(c, name='__nv_int_as_float')
+ !dir$ ignore_tkr (d) i
+ integer, value :: i
+ end function
+ end interface
+
+ interface float_as_int
+ attributes(device) integer function __float_as_int(i) bind(c, name='__nv_float_as_int')
+ !dir$ ignore_tkr (d) i
+ real, value :: i
+ end function
+ end interface
+
interface __float2half_rn
attributes(device) real(2) function __float2half_rn(r) bind(c, name='__nv_float2half_rn')
!dir$ ignore_tkr (d) r
diff --git a/flang/test/Lower/CUDA/cuda-libdevice.cuf b/flang/test/Lower/CUDA/cuda-libdevice.cuf
index f9c5dcc5fc4c3..258e9febfdd6f 100644
--- a/flang/test/Lower/CUDA/cuda-libdevice.cuf
+++ b/flang/test/Lower/CUDA/cuda-libdevice.cuf
@@ -244,3 +244,21 @@ end subroutine
! 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
+
+attributes(global) subroutine test_int_as_float()
+ integer :: i
+ real :: res
+ res = int_as_float(i)
+end subroutine
+
+! CHECK-LABEL: _QPtest_int_as_float
+! CHECK: %{{.*}} = fir.call @__nv_int_as_float(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
+
+attributes(global) subroutine test_float_as_int()
+ integer :: res
+ real :: r
+ res = float_as_int(r)
+end subroutine
+
+! CHECK-LABEL: _QPtest_float_as_int
+! CHECK: %{{.*}} = fir.call @__nv_float_as_int(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f32) -> i32
More information about the flang-commits
mailing list