Skip to content

Commit 9339760

Browse files
authored
Use dynamically sized arrays to avoid leak, thanks to @hhsecond for spotting this out (#135)
1 parent 7b8ae9c commit 9339760

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/backends/torch.c

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,10 @@ int RAI_ModelRunTorch(RAI_ModelRunCtx* mctx, RAI_Error *error) {
5151
size_t ninputs = array_len(mctx->inputs);
5252
size_t noutputs = array_len(mctx->outputs);
5353

54-
DLManagedTensor** inputs = RedisModule_Calloc(ninputs, sizeof(*inputs));
55-
DLManagedTensor** outputs = RedisModule_Calloc(noutputs, sizeof(*outputs));
54+
// DLManagedTensor** inputs = RedisModule_Calloc(ninputs, sizeof(*inputs));
55+
// DLManagedTensor** outputs = RedisModule_Calloc(noutputs, sizeof(*outputs));
56+
DLManagedTensor* inputs[ninputs];
57+
DLManagedTensor* outputs[noutputs];
5658

5759
for (size_t i=0 ; i<ninputs; ++i) {
5860
inputs[i] = &mctx->inputs[i].tensor->tensor;

0 commit comments

Comments
 (0)