1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586 |
- ! Fixed-mode host_data interaction with CUDA BLAS.
- ! { dg-do run { target openacc_nvidia_accel_selected } }
- ! { dg-additional-options "-lcublas -Wall -Wextra" }
- ! { dg-require-effective-target openacc_cublas }
- include "cublas-fixed.h"
- integer, parameter :: N = 10
- integer :: i
- real*4 :: x_ref(N), y_ref(N), x(N), y(N), a
- a = 2.0
- do i = 1, N
- x(i) = 4.0 * i
- y(i) = 3.0
- x_ref(i) = x(i)
- y_ref(i) = y(i)
- end do
- call saxpy (N, a, x_ref, y_ref)
-
- !$acc data copyin (x) copy (y)
- !$acc host_data use_device (x, y)
- call cublassaxpy(N, a, x, 1, y, 1)
- !$acc end host_data
- !$acc end data
- call validate_results (N, y, y_ref)
- !$acc data create (x) copyout (y)
- !$acc parallel loop
- do i = 1, N
- y(i) = 3.0
- end do
- !$acc end parallel loop
- !$acc host_data use_device (x, y)
- call cublassaxpy(N, a, x, 1, y, 1)
- !$acc end host_data
- !$acc end data
- call validate_results (N, y, y_ref)
- y(:) = 3.0
-
- !$acc data copyin (x) copyin (a) copy (y)
- !$acc parallel present (x) pcopy (y) present (a)
- call saxpy (N, a, x, y)
- !$acc end parallel
- !$acc end data
- call validate_results (N, y, y_ref)
- y(:) = 3.0
-
- !$acc enter data copyin (x, a, y)
- !$acc parallel present (x) pcopy (y) present (a)
- call saxpy (N, a, x, y)
- !$acc end parallel
- !$acc exit data delete (x, a) copyout (y)
- call validate_results (N, y, y_ref)
- end
- subroutine saxpy (nn, aa, xx, yy)
- integer :: nn
- real*4 :: aa, xx(nn), yy(nn)
- integer i
- !$acc routine
- do i = 1, nn
- yy(i) = yy(i) + aa * xx(i)
- end do
- end subroutine saxpy
- subroutine validate_results (n, a, b)
- integer :: n
- real*4 :: a(n), b(n)
- do i = 1, N
- if (abs(a(i) - b(i)) > 0.0001) stop 1
- end do
- end subroutine validate_results
|