llvm-project/offload/test/offloading/target_update_ptr_variable_stride.c
Amit Tiwari 14de1bb711
[Clang][OpenMP] Support expression semantics in target update fields with non-contiguous array sections (#176708)
### Issue: Variable stride not recognized as non-contiguous
`CGOpenMPRuntime.cpp` failed to detect `DeclRefExpr`, `MemberExpr`,
`ArraySubscriptExpr` as non-contiguous.

**Fixes**:

`clang/lib/CodeGen/CGOpenMPRuntime.cpp` - Variable stride detection +
dimension count logic
Detect variable stride expressions
(`DeclRefExpr/MemberExpr/ArraySubscriptExpr`) as non-contiguous

Added testcases to cover stack arrays, heap pointers, struct members,
etc., for expression semantics in non-contiguous update.
2026-03-10 19:26:17 +05:30

96 lines
2.1 KiB
C

// RUN: %libomptarget-compile-run-and-check-generic
// Tests non-contiguous array sections with variable stride on heap-allocated
// pointers.
#include <omp.h>
#include <stdio.h>
#include <stdlib.h>
int main() {
int stride = 2;
int len = 10;
double *result = (double *)malloc(len * sizeof(double));
// Initialize
for (int i = 0; i < len; i++) {
result[i] = 0;
}
#pragma omp target enter data map(to : stride, len, result[0 : len])
#pragma omp target map(alloc : result[0 : len])
{
for (int i = 0; i < len; i++) {
result[i] = i;
}
}
// Test FROM
#pragma omp target map(alloc : result[0 : len])
{
for (int i = 0; i < len; i++) {
result[i] += i * 10;
}
}
#pragma omp target update from(result[0 : 5 : stride])
printf("heap ptr variable stride (from):\n");
for (int i = 0; i < len; i++)
printf("%f\n", result[i]);
// Test TO: Reset, modify host, update to device
#pragma omp target map(alloc : result[0 : len])
{
for (int i = 0; i < len; i++) {
result[i] = i * 2;
}
}
for (int i = 0; i < 5; i++) {
result[i * stride] = i + 100;
}
#pragma omp target update to(result[0 : 5 : stride])
#pragma omp target map(alloc : result[0 : len])
{
for (int i = 0; i < len; i++) {
result[i] += 100;
}
}
#pragma omp target update from(result[0 : len])
printf("heap ptr variable stride (to):\n");
for (int i = 0; i < len; i++)
printf("%f\n", result[i]);
#pragma omp target exit data map(delete : stride, len, result[0 : len])
free(result);
return 0;
}
// CHECK: heap ptr variable stride (from):
// CHECK-NEXT: 0.000000
// CHECK-NEXT: 0.000000
// CHECK-NEXT: 22.000000
// CHECK-NEXT: 0.000000
// CHECK-NEXT: 44.000000
// CHECK-NEXT: 0.000000
// CHECK-NEXT: 66.000000
// CHECK-NEXT: 0.000000
// CHECK-NEXT: 88.000000
// CHECK-NEXT: 0.000000
// CHECK: heap ptr variable stride (to):
// CHECK-NEXT: 200.000000
// CHECK-NEXT: 102.000000
// CHECK-NEXT: 201.000000
// CHECK-NEXT: 106.000000
// CHECK-NEXT: 202.000000
// CHECK-NEXT: 110.000000
// CHECK-NEXT: 203.000000
// CHECK-NEXT: 114.000000
// CHECK-NEXT: 204.000000
// CHECK-NEXT: 118.000000