Merge branch 'main' into pytorch_example

msaroufim · web-flow · commit 01ad80840e65 · 2025-04-29T13:13:45.000-07:00
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`	`"""`