Function saxpy

Performs the operation $$y \leftarrow \alpha x + y$$

import { init, cleanup } from "wgblas";
import { saxpy } from "wgblas/saxpy";

const device = await init();

const n = 5;
const alpha = 2;
const x = new Float32Array([1, 2, 3, 4, 5]);
const y = new Float32Array([10, 20, 30, 40, 50]);

console.log("x before:", x);
console.log("y before:", y);
const { y: resultY } = await saxpy(device, n, alpha, x, 1, y, 1);
console.log("y after: ", resultY);
if (typeof process !== "undefined") cleanup();
Copy

Browser (standalone HTML):

<!doctype html>
<html lang="en">
  <head>
    <meta charset="UTF-8" />
    <title>saxpy — wgblas browser example</title>
    <script src="https://unpkg.com/wgblas/dist/wgblas.browser.js"></script>
  </head>
  <body>
    <pre id="out">Running…</pre>
    <script>
      const { init, saxpy, cleanup } = window.wgblas;

      (async () => {
        const device = await init();

        const n = 5;
        const alpha = 2;
        const x = new Float32Array([1, 2, 3, 4, 5]);
        const y = new Float32Array([10, 20, 30, 40, 50]);

        const xBefore = Array.from(x).map(v => v.toFixed(4)).join(", ");
        const yBefore = Array.from(y).map(v => v.toFixed(4)).join(", ");

        const { y: resultY } = await saxpy(device, n, alpha, x, 1, y, 1);

        document.getElementById("out").textContent =
          "x before: " + xBefore +
          "\ny before: " + yBefore +
          "\ny after:  " + Array.from(resultY).map(v => v.toFixed(4)).join(", ");

        cleanup();
      })();
    </script>
  </body>
</html>
Copy

Performs the operation $$y \leftarrow \alpha x + y$$

import { init, cleanup } from "wgblas";
import { saxpy } from "wgblas/saxpy";
import { sscal } from "wgblas/sscal";
import { GpuVector } from "wgblas/classes/GpuVector";

const device = await init();

const n = 5;
const alpha = 2;
const scale = 0.5;
const x = new Float32Array([1, 2, 3, 4, 5]);
const y = new Float32Array([10, 20, 30, 40, 50]);

const xGpu = GpuVector.from(x);
const yGpu = GpuVector.from(y);

console.log("x:      ", x);
console.log("y:      ", y);

// results stay in the GPU.
await saxpy(device, n, alpha, xGpu, 1, yGpu, 1);
await sscal(device, n, scale, yGpu, 1);

// single readback
const result = await yGpu.read();
console.log("result: ", result);

xGpu.destroy();
yGpu.destroy();

if (typeof process !== "undefined") cleanup();
Copy