[flang-commits] [flang] fd3f052 - [flang][cuda] Add interfaces for int_as_float and float_as_int (#153716)

[via flang-commits]

Author: Valentin Clement (バレンタイン クレメン)
Date: 2025-08-15T10:00:53-07:00
New Revision: fd3f052aeb59c1672db7f72169ea4b03c73c62d7

URL: https://github.com/llvm/llvm-project/commit/fd3f052aeb59c1672db7f72169ea4b03c73c62d7
DIFF: https://github.com/llvm/llvm-project/commit/fd3f052aeb59c1672db7f72169ea4b03c73c62d7.diff

LOG: [flang][cuda] Add interfaces for int_as_float and float_as_int (#153716)

Added: 
    

Modified: 
    flang/module/cudadevice.f90
    flang/test/Lower/CUDA/cuda-libdevice.cuf

Removed: 
    


################################################################################
diff  --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90
index e6e818bdd5249..5a7d97b3189be 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 ecc05e79b1cd7..d3167ba69d7cc 100644
--- a/flang/test/Lower/CUDA/cuda-libdevice.cuf
+++ b/flang/test/Lower/CUDA/cuda-libdevice.cuf
@@ -245,6 +245,24 @@ end subroutine
 ! 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
+
 attributes(global) subroutine test_double2loint()
   integer :: res
   double precision :: r