From 9273233a9ccde628f4ec3952ec559100d4cba1b1 Mon Sep 17 00:00:00 2001 From: Valentin Clement Date: Thu, 14 Aug 2025 21:23:27 -0700 Subject: [PATCH] [flang][cuda] Add interface for __fdividef --- flang/module/cudadevice.f90 | 7 +++++++ flang/test/Lower/CUDA/cuda-libdevice.cuf | 9 +++++++++ 2 files changed, 16 insertions(+) diff --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90 index ffc3a3b170ca6..3ffac40c97279 100644 --- a/flang/module/cudadevice.f90 +++ b/flang/module/cudadevice.f90 @@ -399,6 +399,13 @@ attributes(device) subroutine sincospi(x, y, z) bind(c,name='__nv_sincospi') end function end interface + interface __fdividef + attributes(device) real function __fdividef(r,d) bind(c, name='__nv_fast_fdividef') + !dir$ ignore_tkr (d) r, (d) d + real, value :: r,d + end function + end interface + interface __sinf attributes(device) real function __sinf(r) bind(c, name='__nv_sinf') !dir$ ignore_tkr (d) r diff --git a/flang/test/Lower/CUDA/cuda-libdevice.cuf b/flang/test/Lower/CUDA/cuda-libdevice.cuf index f9c5dcc5fc4c3..c560d4c884317 100644 --- a/flang/test/Lower/CUDA/cuda-libdevice.cuf +++ b/flang/test/Lower/CUDA/cuda-libdevice.cuf @@ -244,3 +244,12 @@ end subroutine ! CHECK: %{{.*}} = fir.call @__nv_float2uint_rn(%{{.*}}) proc_attrs fastmath : (f32) -> i32 ! CHECK: %{{.*}} = fir.call @__nv_float2uint_ru(%{{.*}}) proc_attrs fastmath : (f32) -> i32 ! CHECK: %{{.*}} = fir.call @__nv_float2uint_rz(%{{.*}}) proc_attrs fastmath : (f32) -> i32 + +attributes(global) subroutine test_fdividef() + real :: res + real :: r + res = __fdividef(r, r) +end subroutine + +! CHECK-LABEL: _QPtest_fdividef +! CHECK: %{{.*}} = fir.call @__nv_fast_fdividef(%{{.*}}, %{{.*}}) proc_attrs fastmath : (f32, f32) -> f32