| // RUN: %libomptarget-compile-run-and-check-generic | |
| // UNSUPPORTED: amdgcn-amd-amdhsa | |
| #include <assert.h> | |
| #include <stdio.h> | |
| int main() { | |
| int data1[3] = {1, 2, 5}; | |
| int data2[3] = {10, 20, 50}; | |
| int data3[3] = {100, 200, 500}; | |
| int sum[16] = {0}; | |
| for (int i=0; i<16; i++) sum[i] = 10000; | |
| #pragma omp target teams distribute parallel for map(tofrom : sum[:16]) \ | |
| firstprivate(data1, data2, data3) | |
| for (int i = 0; i < 16; ++i) { | |
| for (int j = 0; j < 3; ++j) { | |
| sum[i] += data1[j]; | |
| sum[i] += data2[j]; | |
| sum[i] += data3[j]; | |
| } | |
| } | |
| int correct = 1; | |
| for (int i = 0; i < 16; ++i) { | |
| if (sum[i] != 10888) { | |
| correct = 0; | |
| printf("ERROR: The sum for index %d is %d\n", i, sum[i]); | |
| printf("ERROR: data1 = {%d, %d, %d}\n", data1[0], data1[1], data1[2]); | |
| printf("ERROR: data2 = {%d, %d, %d}\n", data2[0], data2[1], data2[2]); | |
| printf("ERROR: data3 = {%d, %d, %d}\n", data3[0], data3[1], data3[2]); | |
| break; | |
| } | |
| } | |
| fflush(stdout); | |
| assert(correct); | |
| printf("PASS\n"); | |
| return 0; | |
| } | |
| // CHECK: PASS |