fix

unixpickle · unixpickle · commit 36d1abe9efdc · 2024-03-18T00:41:36.000-04:00
diff --git a/accelerated_scan/warp.cuh b/accelerated_scan/warp.cuh
@@ -355,7 +355,7 @@ __device__ Tuple load_shifted_tuple(const Tuple* ptr, int index, int limit) {
     for (int i = 0; i < Tuple::Size; i++) {
         const int idx = index * Tuple::Size + i + offset;
         if (idx >= 0 && idx < limit * Tuple::Size) {
-            x.data[i] = rawPtr[offset];
+            x.data[i] = rawPtr[idx];
         } else {
             x.data[i] = 0.0;
         }

Original file line number	Diff line number	Diff line change
`@@ -355,7 +355,7 @@ __device__ Tuple load_shifted_tuple(const Tuple* ptr, int index, int limit) {`
`355`	`355`	`for (int i = 0; i < Tuple::Size; i++) {`
`356`	`356`	`const int idx = index * Tuple::Size + i + offset;`
`357`	`357`	`if (idx >= 0 && idx < limit * Tuple::Size) {`
`358`		`- x.data[i] = rawPtr[offset];`
	`358`	`+ x.data[i] = rawPtr[idx];`
`359`	`359`	`} else {`
`360`	`360`	`x.data[i] = 0.0;`
`361`	`361`	`}`