Skip to content

Commit

Permalink
Merge branch 'branch-24.10' into tb
Browse files Browse the repository at this point in the history
  • Loading branch information
galipremsagar committed Sep 23, 2024
2 parents b806121 + 0400d7a commit a36580a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/cuml/cuml/fil/fil.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ rmm = gpu_only_import('rmm')

from libcpp cimport bool
from libc.stdint cimport uintptr_t
from libc.stdlib cimport free
from libc.stdlib cimport free as c_free

import cuml.internals
from cuml.internals.array import CumlArray
Expand Down Expand Up @@ -545,7 +545,7 @@ cdef class ForestInference_impl():
treelite_params.threads_per_tree = kwargs['threads_per_tree']
if kwargs['compute_shape_str']:
if self.shape_str:
free(self.shape_str)
c_free(self.shape_str)
treelite_params.pforest_shape_str = &self.shape_str
else:
treelite_params.pforest_shape_str = NULL
Expand Down

0 comments on commit a36580a

Please sign in to comment.