Skip to content

Commit bf7e37f

Browse files
fix: failing tests (#2886)
Co-authored-by: Angus Hollands <goosey15@gmail.com>
1 parent b32c86a commit bf7e37f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/awkward/_connect/cuda/cuda_kernels/awkward_ByteMaskedArray_getitem_nextcarry.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@ template <typename T, typename C>
1515
__global__ void
1616
awkward_ByteMaskedArray_getitem_nextcarry_a(T* tocarry,
1717
const C* mask,
18-
bool validwhen,
1918
int64_t length,
19+
bool validwhen,
2020
int64_t* scan_in_array,
2121
uint64_t invocation_index,
2222
uint64_t* err_code) {
@@ -37,8 +37,8 @@ template <typename T, typename C>
3737
__global__ void
3838
awkward_ByteMaskedArray_getitem_nextcarry_b(T* tocarry,
3939
const C* mask,
40-
bool validwhen,
4140
int64_t length,
41+
bool validwhen,
4242
int64_t* scan_in_array,
4343
uint64_t invocation_index,
4444
uint64_t* err_code) {

src/awkward/_connect/cuda/cuda_kernels/awkward_ListArray_getitem_jagged_expand.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ awkward_ListArray_getitem_jagged_expand(T* multistarts,
1818
uint64_t invocation_index,
1919
uint64_t* err_code) {
2020
if (err_code[0] == NO_ERROR) {
21-
int64_t thread_id = (blockIdx.x * blockDim.x + threadIdx.x) / jaggedsize;
21+
int64_t thread_id = (blockIdx.x * blockDim.x + threadIdx.x) % length;
2222
int64_t thready_id = (blockIdx.x * blockDim.x + threadIdx.x) % jaggedsize;
2323
W start = fromstarts[thread_id];
2424
X stop = fromstops[thread_id];

0 commit comments

Comments
 (0)