fix indexing bug in saxpy.py (NVIDIA#582)

msaroufim · leofang · web-flow · commit a3cd5aebef7a · 2025-04-29T15:38:17.000-04:00
* fix indexing bug in saxpy.py

* signoff

Signed-off-by: Mark Saroufim &lt;marksaroufim@meta.com&gt;

---------

Signed-off-by: Mark Saroufim &lt;marksaroufim@meta.com&gt;
Co-authored-by: Leo Fang &lt;leof@nvidia.com&gt;
diff --git a/cuda_core/examples/saxpy.py b/cuda_core/examples/saxpy.py
@@ -18,7 +18,7 @@
                       size_t N) {
     const unsigned int tid = threadIdx.x + blockIdx.x * blockDim.x;
     for (size_t i=tid; i<N; i+=gridDim.x*blockDim.x) {
-        out[tid] = a * x[tid] + y[tid];
+        out[i] = a * x[i] + y[i];
     }
 }
 """

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@`
`18`	`18`	`size_t N) {`
`19`	`19`	`const unsigned int tid = threadIdx.x + blockIdx.x * blockDim.x;`
`20`	`20`	`for (size_t i=tid; i<N; i+=gridDim.x*blockDim.x) {`
`21`		`- out[tid] = a * x[tid] + y[tid];`
	`21`	`+ out[i] = a * x[i] + y[i];`
`22`	`22`	`}`
`23`	`23`	`}`
`24`	`24`	`"""`