From fe465d0274ffd26248fd6b0d74edaf439834ff86 Mon Sep 17 00:00:00 2001 From: Valentin Clement Date: Thu, 14 Aug 2025 12:55:54 -0700 Subject: [PATCH] [flang][cuda] Add bind names for __double2ll_rX interfaces --- flang/module/cudadevice.f90 | 14 +++++++------- flang/test/Lower/CUDA/cuda-libdevice.cuf | 15 +++++++++++++++ 2 files changed, 22 insertions(+), 7 deletions(-) diff --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90 index 432125097e9d2..1e403b1ffdb91 100644 --- a/flang/module/cudadevice.f90 +++ b/flang/module/cudadevice.f90 @@ -625,29 +625,29 @@ attributes(device) real(8) function sinpi(x) bind(c,name='__nv_sinpi') end function end interface - interface __double2ll_rn - attributes(device) integer(8) function __double2ll_rn(r) bind(c) + interface __double2ll_rd + attributes(device) integer(8) function __double2ll_rd(r) bind(c, name='__nv_double2ll_rd') !dir$ ignore_tkr (d) r double precision, value :: r end function end interface - interface __double2ll_rz - attributes(device) integer(8) function __double2ll_rz(r) bind(c) + interface __double2ll_rn + attributes(device) integer(8) function __double2ll_rn(r) bind(c, name='__nv_double2ll_rn') !dir$ ignore_tkr (d) r double precision, value :: r end function end interface interface __double2ll_ru - attributes(device) integer(8) function __double2ll_ru(r) bind(c) + attributes(device) integer(8) function __double2ll_ru(r) bind(c, name='__nv_double2ll_ru') !dir$ ignore_tkr (d) r double precision, value :: r end function end interface - interface __double2ll_rd - attributes(device) integer(8) function __double2ll_rd(r) bind(c) + interface __double2ll_rz + attributes(device) integer(8) function __double2ll_rz(r) bind(c, name='__nv_double2ll_rz') !dir$ ignore_tkr (d) r double precision, value :: r end function diff --git a/flang/test/Lower/CUDA/cuda-libdevice.cuf b/flang/test/Lower/CUDA/cuda-libdevice.cuf index 6b80fb2e74146..844bdb954924a 100644 --- a/flang/test/Lower/CUDA/cuda-libdevice.cuf +++ b/flang/test/Lower/CUDA/cuda-libdevice.cuf @@ -100,3 +100,18 @@ end subroutine ! CHECK-LABEL: _QPtest_exp ! CHECK: %{{.*}} = fir.call @__nv_expf(%{{.*}}) proc_attrs fastmath : (f32) -> f32 ! CHECK: %{{.*}} = fir.call @__nv_exp10f(%{{.*}}) proc_attrs fastmath : (f32) -> f32 + +attributes(global) subroutine test_double2ll_rX() + integer(8) :: res + double precision :: r + res = __double2ll_rd(r) + res = __double2ll_rn(r) + res = __double2ll_ru(r) + res = __double2ll_rz(r) +end subroutine + +! CHECK-LABEL: _QPtest_double2ll_rx +! CHECK: %{{.*}} = fir.call @__nv_double2ll_rd(%{{.*}}) proc_attrs fastmath : (f64) -> i64 +! CHECK: %{{.*}} = fir.call @__nv_double2ll_rn(%{{.*}}) proc_attrs fastmath : (f64) -> i64 +! CHECK: %{{.*}} = fir.call @__nv_double2ll_ru(%{{.*}}) proc_attrs fastmath : (f64) -> i64 +! CHECK: %{{.*}} = fir.call @__nv_double2ll_rz(%{{.*}}) proc_attrs fastmath : (f64) -> i64