diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-02-18 17:59:43 +0100 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-02-18 17:59:43 +0100 |
commit | 76c21a95c29bd3645213b870de61e012204cc844 (patch) | |
tree | 31e5bc235d917bcd930c5e600d7aa450742d64a5 /src/pyclblast/samples/saxpy.py | |
parent | a66e24a009fe6f23c6231ec3b7c1a4698f831435 (diff) |
Added PyCLBlast samples
Diffstat (limited to 'src/pyclblast/samples/saxpy.py')
-rw-r--r-- | src/pyclblast/samples/saxpy.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/pyclblast/samples/saxpy.py b/src/pyclblast/samples/saxpy.py new file mode 100644 index 00000000..098e44d5 --- /dev/null +++ b/src/pyclblast/samples/saxpy.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python + +# This file is part of the CLBlast project. The project is licensed under Apache Version 2.0. +# This file follows the PEP8 Python style guide and uses a max-width of 100 characters per line. +# +# Author(s): +# Cedric Nugteren <www.cedricnugteren.nl> + +import numpy as np +import pyopencl as cl +from pyopencl.array import Array +import pyclblast + +# Settings for this sample +dtype = 'float32' +alpha = 1.5 +n = 4 + +print("# Setting up OpenCL") +ctx = cl.create_some_context() +queue = cl.CommandQueue(ctx) + +print("# Setting up Numpy arrays") +x = np.random.rand(n).astype(dtype=dtype) +y = np.random.rand(n).astype(dtype=dtype) + +print("# Setting up OpenCL arrays") +clx = Array(queue, x.shape, x.dtype) +cly = Array(queue, y.shape, y.dtype) +clx.set(x) +cly.set(y) + +print("# Example level-1 operation: AXPY") +pyclblast.axpy(queue, n, clx, cly, alpha=alpha) +print("# Result for vector y: %s" % cly.get()) +print("# Expected result: %s" % (alpha * x + y)) |