Skip to content

Commit 27e5799

Browse files
committed
Fix initialization of strides
1 parent 1691384 commit 27e5799

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/tensor.c

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,11 +253,14 @@ RAI_Tensor* RAI_TensorCreate(const char* dataTypeStr, long long* dims, int ndims
253253
int64_t* shape = RedisModule_Calloc(ndims, sizeof(*shape));
254254
int64_t* strides = RedisModule_Calloc(ndims, sizeof(*strides));
255255
size_t len = 1;
256-
for (long long i = 0 ; i < ndims ; ++i){
256+
for (int64_t i = 0 ; i < ndims ; ++i){
257257
shape[i] = dims[i];
258258
strides[i] = 1;
259259
len *= dims[i];
260260
}
261+
for (int64_t i = ndims-2 ; i >= 0 ; --i) {
262+
strides[i] *= strides[i+1] * shape[i+1];
263+
}
261264

262265
DLContext ctx = (DLContext){
263266
.device_type = kDLCPU,

0 commit comments

Comments
 (0)