Skip to content

Commit 16775f2

Browse files
committed
merge with master
1 parent 076b70a commit 16775f2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/tensor.c

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,10 @@ RAI_Tensor *RAI_TensorCreateWithDLDataType(DLDataType dtype, long long *dims, in
122122
strides[i] *= strides[i + 1] * shape[i + 1];
123123
}
124124

125-
DLContext ctx = (DLContext){.device_type = kDLCPU, .device_id = 0};
125+
DLDevice device = (DLDevice){.device_type = kDLCPU, .device_id = 0};
126126
void *data = RedisModule_Calloc(len, dtypeSize);
127127

128-
ret->tensor = (DLManagedTensor){.dl_tensor = (DLTensor){.ctx = ctx,
128+
ret->tensor = (DLManagedTensor){.dl_tensor = (DLTensor){.device = device,
129129
.data = data,
130130
.ndim = ndims,
131131
.dtype = dtype,
@@ -170,7 +170,7 @@ RAI_Tensor *_TensorCreateWithDLDataTypeAndRString(DLDataType dtype, size_t dtype
170170
strides[i] *= strides[i + 1] * shape[i + 1];
171171
}
172172

173-
DLContext ctx = (DLContext){.device_type = kDLCPU, .device_id = 0};
173+
DLDevice device = (DLDevice){.device_type = kDLCPU, .device_id = 0};
174174
size_t nbytes = len * dtypeSize;
175175

176176
size_t blob_len;
@@ -186,7 +186,7 @@ RAI_Tensor *_TensorCreateWithDLDataTypeAndRString(DLDataType dtype, size_t dtype
186186
RAI_HoldString(NULL, rstr);
187187

188188
RAI_Tensor *ret = RAI_TensorNew();
189-
ret->tensor = (DLManagedTensor){.dl_tensor = (DLTensor){.ctx = ctx,
189+
ret->tensor = (DLManagedTensor){.dl_tensor = (DLTensor){.device = device,
190190
.data = data,
191191
.ndim = ndims,
192192
.dtype = dtype,
@@ -327,7 +327,7 @@ RAI_Tensor *RAI_TensorCreateFromDLTensor(DLManagedTensor *dl_tensor) {
327327
RAI_Tensor *ret = RAI_TensorNew();
328328

329329
ret->tensor =
330-
(DLManagedTensor){.dl_tensor = (DLTensor){.ctx = dl_tensor->dl_tensor.ctx,
330+
(DLManagedTensor){.dl_tensor = (DLTensor){.device = dl_tensor->dl_tensor.device,
331331
.data = dl_tensor->dl_tensor.data,
332332
.ndim = dl_tensor->dl_tensor.ndim,
333333
.dtype = dl_tensor->dl_tensor.dtype,

0 commit comments

Comments
 (0)