mxnet-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From TongKe Xue <tk...@tkxue.org>
Subject Re: disposing all ndarray in a given context
Date Tue, 17 Oct 2017 14:18:08 GMT
Following up to this:

I see that the Scala API, when creating ndarray, uses:

https://github.com/apache/incubator-mxnet/blob/master/scala-package/core/src/main/scala/ml/dmlc/mxnet/NDArray.scala#L114

which calls

https://github.com/apache/incubator-mxnet/blob/master/scala-package/core/src/main/scala/ml/dmlc/mxnet/LibInfo.scala#L42

to get a "handle" from the given context.


I've looked through the LibInfo.scala file -- and it's not clear to me
if there is a way to:

1) nuke all handles in a Context OR
2) get a list of all handles in a Context (so I can manually call dispose)

Is either of these things possible?

Thanks!


On Mon, Oct 16, 2017 at 4:15 PM, TongKe Xue <tkxue@tkxue.org> wrote:
> Quoting: https://github.com/apache/incubator-mxnet/blob/master/scala-package/core/src/main/scala/ml/dmlc/mxnet/NDArray.scala#L545-L546
>
> * WARNING: it is your responsibility to clear this object through dispose().
> * NEVER rely on the GC strategy
>
> Is there a way to say "dispose all ndarrays of this context" ?

Mime
View raw message