[flang-commits] [flang] [flang][cuda] Add bind(c) interfaces for __fadd_rd and __fadd_ru (PR #122535)
via flang-commits
flang-commits at lists.llvm.org
Fri Jan 10 13:49:25 PST 2025
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-flang-fir-hlfir
Author: Valentin Clement (バレンタイン クレメン) (clementval)
<details>
<summary>Changes</summary>
Function like `__fadd_rd ` and `__fadd_ru ` need to be converted to the cuda equivalent.
---
Full diff: https://github.com/llvm/llvm-project/pull/122535.diff
2 Files Affected:
- (modified) flang/module/cudadevice.f90 (+14)
- (added) flang/test/Lower/CUDA/cuda-intrinsic.cuf (+17)
``````````diff
diff --git a/flang/module/cudadevice.f90 b/flang/module/cudadevice.f90
index 1402bd4e150419..74825c36f7a95f 100644
--- a/flang/module/cudadevice.f90
+++ b/flang/module/cudadevice.f90
@@ -71,4 +71,18 @@ attributes(device) subroutine threadfence_system()
end interface
public :: threadfence_system
+ interface
+ attributes(device) function __fadd_rd(x, y) bind(c, name='__nv_fadd_rd')
+ real :: x, y, __fadd_rd
+ end function
+ end interface
+ public :: __fadd_rd
+
+ interface
+ attributes(device) function __fadd_ru(x, y) bind(c, name='__nv_fadd_ru')
+ real :: x, y, __fadd_ru
+ end function
+ end interface
+ public :: __fadd_ru
+
end module
diff --git a/flang/test/Lower/CUDA/cuda-intrinsic.cuf b/flang/test/Lower/CUDA/cuda-intrinsic.cuf
new file mode 100644
index 00000000000000..9723afc532387f
--- /dev/null
+++ b/flang/test/Lower/CUDA/cuda-intrinsic.cuf
@@ -0,0 +1,17 @@
+! RUN: bbc -emit-hlfir -fcuda %s -o - | FileCheck %s
+
+module mod1
+ type int
+ real :: inf, sup
+ end type int
+contains
+ attributes(global) subroutine fadd(c, a, b)
+ type (int) :: c, a, b
+ c%inf = __fadd_rd(a%inf, b%inf)
+ c%sup = __fadd_ru(a%sup, b%sup)
+ end subroutine
+end
+
+! CHECK-LABEL: func.func @_QMmod1Pfadd
+! CHECK: fir.call @__nv_fadd_rd
+! CHECK: fir.call @__nv_fadd_ru
``````````
</details>
https://github.com/llvm/llvm-project/pull/122535
More information about the flang-commits
mailing list