From b940c6da7e2d74aa89f456bdfd0cd5dfc4d112c7 Mon Sep 17 00:00:00 2001 From: Svyatoslav Scherbina Date: Wed, 22 Jul 2020 19:24:37 +0300 Subject: [PATCH] torch sample: workaround some of double-dispose bugs --- samples/torch/src/torchMain/kotlin/Tensors.kt | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/samples/torch/src/torchMain/kotlin/Tensors.kt b/samples/torch/src/torchMain/kotlin/Tensors.kt index c739f95ef5e..c7097ef7e88 100644 --- a/samples/torch/src/torchMain/kotlin/Tensors.kt +++ b/samples/torch/src/torchMain/kotlin/Tensors.kt @@ -34,7 +34,11 @@ abstract class FloatTensor(val raw: CPointer) : Disposable { fun sum() = THFloatTensor_sumall(raw) fun flatten() = (0 until elements.size).map { data[it] }.toTypedArray() + private var disposed = false override fun dispose() { + if (disposed) return + disposed = true + THFloatTensor_free(raw) }