Skip to content

Commit 3bc4d66

Browse files
authored
[flang][cuda] Add interfaces for __int2float_rX (#153708)
1 parent abe92a5 commit 3bc4d66

File tree

2 files changed

+44
-0
lines changed

2 files changed

+44
-0
lines changed

flang/module/cudadevice.f90

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -674,6 +674,34 @@ attributes(device) real(8) function sinpi(x) bind(c,name='__nv_sinpi')
674674
end function
675675
end interface
676676

677+
interface __int2float_rd
678+
attributes(device) real function __int2float_rd(i) bind(c, name='__nv_int2float_rd')
679+
!dir$ ignore_tkr (d) i
680+
integer, value :: i
681+
end function
682+
end interface
683+
684+
interface __int2float_rn
685+
attributes(device) real function __int2float_rn(i) bind(c, name='__nv_int2float_rn')
686+
!dir$ ignore_tkr (d) i
687+
integer, value :: i
688+
end function
689+
end interface
690+
691+
interface __int2float_ru
692+
attributes(device) real function __int2float_ru(i) bind(c, name='__nv_int2float_ru')
693+
!dir$ ignore_tkr (d) i
694+
integer, value :: i
695+
end function
696+
end interface
697+
698+
interface __int2float_rz
699+
attributes(device) real function __int2float_rz(i) bind(c, name='__nv_int2float_rz')
700+
!dir$ ignore_tkr (d) i
701+
integer, value :: i
702+
end function
703+
end interface
704+
677705
interface __int2double_rn
678706
attributes(device) double precision function __int2double_rn(i) bind(c)
679707
!dir$ ignore_tkr (d) i

flang/test/Lower/CUDA/cuda-libdevice.cuf

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,22 @@ end subroutine
199199
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
200200
! CHECK: %{{.*}} = fir.call @__nv_ll2float_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
201201
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
202+
203+
attributes(global) subroutine test_int2float_rX()
204+
real :: res
205+
integer :: i
206+
res = __int2float_rd(i)
207+
res = __int2float_rn(i)
208+
res = __int2float_ru(i)
209+
res = __int2float_rz(i)
210+
end subroutine
211+
212+
! CHECK-LABEL: _QPtest_int2float_rx
213+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rd(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
214+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
215+
! CHECK: %{{.*}} = fir.call @__nv_int2float_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
216+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
217+
202218
attributes(global) subroutine test_float2int_rX()
203219
integer :: res
204220
real :: r

0 commit comments

Comments
 (0)