forked from kscience/kmath
disposable pattern
This commit is contained in:
parent
cadcb9916f
commit
78b1cd41da
@ -15,7 +15,7 @@ plugins {
|
||||
description = "Wrapper for the Bayesian Computation library NOA on top of LibTorch"
|
||||
|
||||
dependencies {
|
||||
implementation(project(":kmath-tensors"))
|
||||
api(project(":kmath-tensors"))
|
||||
}
|
||||
|
||||
val home: String = System.getProperty("user.home")
|
||||
|
@ -5,7 +5,7 @@
|
||||
|
||||
package space.kscience.kmath.noa;
|
||||
|
||||
public class JNoa {
|
||||
class JNoa {
|
||||
|
||||
static {
|
||||
String jNoaPath = System.getProperty("user.home") +
|
||||
@ -30,5 +30,8 @@ public class JNoa {
|
||||
public static native void setNumThreads(int numThreads);
|
||||
|
||||
public static native void setSeed(int seed);
|
||||
|
||||
public static native void disposeTensor(long tensorHandle);
|
||||
|
||||
}
|
||||
|
||||
|
@ -0,0 +1,8 @@
|
||||
/*
|
||||
* Copyright 2018-2021 KMath contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package space.kscience.kmath.noa
|
||||
|
||||
public sealed class NoaAlgebra<T>{}
|
@ -0,0 +1,15 @@
|
||||
/*
|
||||
* Copyright 2018-2021 KMath contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package space.kscience.kmath.noa.memory
|
||||
|
||||
public abstract class NoaResource
|
||||
internal constructor(scope: NoaScope) {
|
||||
init {
|
||||
scope.add(::dispose)
|
||||
}
|
||||
|
||||
protected abstract fun dispose(): Unit
|
||||
}
|
@ -0,0 +1,34 @@
|
||||
/*
|
||||
* Copyright 2018-2021 KMath contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package space.kscience.kmath.noa.memory
|
||||
|
||||
private typealias Disposable = () -> Unit
|
||||
|
||||
public class NoaScope {
|
||||
|
||||
private val disposables: ArrayDeque<Disposable> = ArrayDeque(0)
|
||||
|
||||
public fun disposeAll() {
|
||||
disposables.forEach(Disposable::invoke)
|
||||
disposables.clear()
|
||||
}
|
||||
|
||||
internal inline fun add(crossinline disposable: Disposable) {
|
||||
disposables += {
|
||||
try {
|
||||
disposable()
|
||||
} catch (ignored: Throwable) {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
internal inline fun <R> withNoaScope(i: Int, block: NoaScope.() -> R): R {
|
||||
val noaScope = NoaScope()
|
||||
val result = noaScope.block()
|
||||
noaScope.disposeAll()
|
||||
return result
|
||||
}
|
@ -0,0 +1,23 @@
|
||||
/*
|
||||
* Copyright 2018-2021 KMath contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package space.kscience.kmath.noa
|
||||
|
||||
import space.kscience.kmath.noa.memory.NoaResource
|
||||
import space.kscience.kmath.noa.memory.NoaScope
|
||||
import space.kscience.kmath.tensors.api.Tensor
|
||||
|
||||
internal typealias TensorHandle = Long
|
||||
|
||||
public sealed class NoaTensor<T>
|
||||
constructor(scope: NoaScope, internal val tensorHandle: TensorHandle) :
|
||||
NoaResource(scope){
|
||||
|
||||
override fun dispose(): Unit = JNoa.disposeTensor(tensorHandle)
|
||||
}
|
||||
|
||||
public class NoaDoubleTensor
|
||||
internal constructor(scope: NoaScope, tensorHandle: TensorHandle) :
|
||||
NoaTensor<Double>(scope, tensorHandle)
|
@ -47,6 +47,14 @@ JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_setNumThreads
|
||||
JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_setSeed
|
||||
(JNIEnv *, jclass, jint);
|
||||
|
||||
/*
|
||||
* Class: space_kscience_kmath_noa_JNoa
|
||||
* Method: disposeTensor
|
||||
* Signature: (J)V
|
||||
*/
|
||||
JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_disposeTensor
|
||||
(JNIEnv *, jclass, jlong);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
@ -5,6 +5,8 @@
|
||||
|
||||
package space.kscience.kmath.noa
|
||||
|
||||
import space.kscience.kmath.noa.memory.NoaScope
|
||||
import space.kscience.kmath.noa.memory.withNoaScope
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
|
||||
@ -26,4 +28,6 @@ class TestUtils {
|
||||
setNumThreads(numThreads)
|
||||
assertEquals(numThreads, getNumThreads())
|
||||
}
|
||||
|
||||
|
||||
}
|
Loading…
Reference in New Issue
Block a user