From cb0fa67da71f04c3bb767f3f936d8ccc4f467d86 Mon Sep 17 00:00:00 2001 From: Richard Boyd <58230111+richardhboyd@users.noreply.github.com> Date: Mon, 4 Aug 2025 20:55:05 -0400 Subject: [PATCH] fix getting started example --- cuda_core/docs/source/getting-started.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cuda_core/docs/source/getting-started.md b/cuda_core/docs/source/getting-started.md index d27edb2c4..6fffa364e 100644 --- a/cuda_core/docs/source/getting-started.md +++ b/cuda_core/docs/source/getting-started.md @@ -49,6 +49,7 @@ and a corresponding {class}`Stream `. Don't forget to use {meth}`Device.set_current() `! ```python +import cupy as cp from cuda.core.experimental import Device, LaunchConfig, Program, ProgramOptions, launch dev = Device() @@ -76,6 +77,7 @@ ker = mod.get_kernel("vector_add") # Prepare input/output arrays (using CuPy) size = 50000 +rng = cp.random.default_rng() a = rng.random(size, dtype=cp.float32) b = rng.random(size, dtype=cp.float32) c = cp.empty_like(a)