derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
60 lines
1.5 KiB
C
60 lines
1.5 KiB
C
// clang-format off
|
|
// This test verifies correctness of Xteam Reduction for reduced precision types.
|
|
//
|
|
// RUN: %libomptarget-compile-generic -fopenmp-target-fast -lmlir_float16_utils
|
|
// RUN: env LIBOMPTARGET_KERNEL_TRACE=1 %libomptarget-run-generic 2>&1 | %fcheck-generic
|
|
|
|
// UNSUPPORTED: nvptx64-nvidia-cuda
|
|
// UNSUPPORTED: nvptx64-nvidia-cuda-LTO
|
|
// UNSUPPORTED: aarch64-unknown-linux-gnu
|
|
// UNSUPPORTED: aarch64-unknown-linux-gnu-LTO
|
|
// UNSUPPORTED: x86_64-unknown-linux-gnu
|
|
// UNSUPPORTED: x86_64-unknown-linux-gnu-LTO
|
|
|
|
// clang-format on
|
|
#include <omp.h>
|
|
#include <stdio.h>
|
|
|
|
int main() {
|
|
int N = 100;
|
|
|
|
_Float16 a[N];
|
|
__bf16 b[N];
|
|
short c[N];
|
|
|
|
for (int i = 0; i < N; i++) {
|
|
a[i] = i;
|
|
b[i] = i;
|
|
c[i] = i;
|
|
}
|
|
|
|
_Float16 sum1 = 0;
|
|
__bf16 sum2 = 0;
|
|
short sum3 = 0;
|
|
|
|
#pragma omp target teams distribute parallel for map(tofrom:sum1) reduction(+:sum1)
|
|
for (int j = 0; j < N; j = j + 1)
|
|
sum1 += a[j];
|
|
|
|
#pragma omp target teams distribute parallel for map(tofrom:sum2) reduction(+:sum2)
|
|
for (int j = 0; j < N; j = j + 2)
|
|
sum2 += b[j];
|
|
|
|
#pragma omp target teams distribute parallel for map(tofrom:sum3) reduction(+:sum3)
|
|
for (int j = 0; j < N; j = j + 2)
|
|
sum3 += c[j];
|
|
|
|
printf("%f %f %d\n", (float)sum1, (float)sum2, sum3);
|
|
|
|
int rc = (sum1 != 4952) || (sum2 != 2448) || (sum3 != 2450);
|
|
|
|
if (!rc)
|
|
printf("Success\n");
|
|
|
|
return rc;
|
|
}
|
|
|
|
/// CHECK: DEVID:[[S:[ ]*]][[DEVID:[0-9]+]] SGN:8
|
|
/// CHECK: DEVID:[[S:[ ]*]][[DEVID:[0-9]+]] SGN:8
|
|
/// CHECK: DEVID:[[S:[ ]*]][[DEVID:[0-9]+]] SGN:8
|