Skip to content

Commit 5d14255

Browse files
committed
[flang][cuda] Add interfaces for __int2float_rX
1 parent ca9ddd5 commit 5d14255

File tree

2 files changed

+43
-0
lines changed

2 files changed

+43
-0
lines changed

flang/module/cudadevice.f90

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

586+
interface __int2float_rd
587+
attributes(device) real function __int2float_rd(i) bind(c, name='__nv_int2float_rd')
588+
!dir$ ignore_tkr (d) i
589+
integer, value :: i
590+
end function
591+
end interface
592+
593+
interface __int2float_rn
594+
attributes(device) real function __int2float_rn(i) bind(c, name='__nv_int2float_rn')
595+
!dir$ ignore_tkr (d) i
596+
integer, value :: i
597+
end function
598+
end interface
599+
600+
interface __int2float_ru
601+
attributes(device) real function __int2float_ru(i) bind(c, name='__nv_int2float_ru')
602+
!dir$ ignore_tkr (d) i
603+
integer, value :: i
604+
end function
605+
end interface
606+
607+
interface __int2float_rz
608+
attributes(device) real function __int2float_rz(i) bind(c, name='__nv_int2float_rz')
609+
!dir$ ignore_tkr (d) i
610+
integer, value :: i
611+
end function
612+
end interface
613+
586614
interface __int2double_rn
587615
attributes(device) double precision function __int2double_rn(i) bind(c)
588616
!dir$ ignore_tkr (d) i

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

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,3 +176,18 @@ end subroutine
176176
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
177177
! CHECK: %{{.*}} = fir.call @__nv_ll2float_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
178178
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
179+
180+
attributes(global) subroutine test_int2float_rX()
181+
real :: res
182+
integer :: i
183+
res = __int2float_rd(i)
184+
res = __int2float_rn(i)
185+
res = __int2float_ru(i)
186+
res = __int2float_rz(i)
187+
end subroutine
188+
189+
! CHECK-LABEL: _QPtest_int2float_rx
190+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rd(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
191+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
192+
! CHECK: %{{.*}} = fir.call @__nv_int2float_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32
193+
! CHECK: %{{.*}} = fir.call @__nv_int2float_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i32) -> f32

0 commit comments

Comments
 (0)