Skip to content

Commit ca9ddd5

Browse files
authored
[flang][cuda] Add interfaces for __ll2float_rX (#153694)
1 parent fc4df2c commit ca9ddd5

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
@@ -681,6 +681,34 @@ attributes(device) real(8) function sinpi(x) bind(c,name='__nv_sinpi')
681681
end function
682682
end interface
683683

684+
interface __ll2float_rd
685+
attributes(device) real function __ll2float_rd(i) bind(c, name='__nv_ll2float_rd')
686+
!dir$ ignore_tkr (d) i
687+
integer(8), value :: i
688+
end function
689+
end interface
690+
691+
interface __ll2float_rn
692+
attributes(device) real function __ll2float_rn(i) bind(c, name='__nv_ll2float_rn')
693+
!dir$ ignore_tkr (d) i
694+
integer(8), value :: i
695+
end function
696+
end interface
697+
698+
interface __ll2float_ru
699+
attributes(device) real function __ll2float_ru(i) bind(c, name='__nv_ll2float_ru')
700+
!dir$ ignore_tkr (d) i
701+
integer(8), value :: i
702+
end function
703+
end interface
704+
705+
interface __ll2float_rz
706+
attributes(device) real function __ll2float_rz(i) bind(c, name='__nv_ll2float_rz')
707+
!dir$ ignore_tkr (d) i
708+
integer(8), value :: i
709+
end function
710+
end interface
711+
684712
interface __ll2double_rd
685713
attributes(device) double precision function __ll2double_rd(i) bind(c, name='__nv_ll2double_rd')
686714
!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
@@ -161,3 +161,18 @@ end subroutine
161161
! CHECK: %{{.*}} = fir.call @__nv_double2ull_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
162162
! CHECK: %{{.*}} = fir.call @__nv_double2ull_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
163163
! CHECK: %{{.*}} = fir.call @__nv_double2ull_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (f64) -> i64
164+
165+
attributes(global) subroutine test_ll2float_rX()
166+
real :: res
167+
integer(8) :: i
168+
res = __ll2float_rd(i)
169+
res = __ll2float_rn(i)
170+
res = __ll2float_ru(i)
171+
res = __ll2float_rz(i)
172+
end subroutine
173+
174+
! CHECK-LABEL: _QPtest_ll2float_rx
175+
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rd(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
176+
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rn(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
177+
! CHECK: %{{.*}} = fir.call @__nv_ll2float_ru(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32
178+
! CHECK: %{{.*}} = fir.call @__nv_ll2float_rz(%{{.*}}) proc_attrs<bind_c> fastmath<contract> : (i64) -> f32

0 commit comments

Comments
 (0)