[flang][cuda] Add runtime check for passing device arrays (#144003)

2025-06-12 20:47:58 -07:00
parent 07dad4ecba
commit 9992668404
8 changed files with 73 additions and 0 deletions
--- a/flang-rt/lib/cuda/descriptor.cpp
+++ b/flang-rt/lib/cuda/descriptor.cpp
@@ -54,6 +54,14 @@ void RTDEF(CUFSyncGlobalDescriptor)(
  ((Descriptor *)devAddr, (Descriptor *)hostPtr, sourceFile, sourceLine);
 }

+void RTDEF(CUFDescriptorCheckSection)(
+    const Descriptor *desc, const char *sourceFile, int sourceLine) {
+  if (desc && !desc->IsContiguous()) {
+    Terminator terminator{sourceFile, sourceLine};
+    terminator.Crash("device array section argument is not contiguous");
+  }
+}
+
 RT_EXT_API_GROUP_END
 }
 } // namespace Fortran::runtime::cuda