teldufalsari/kmath
2
0
Fork 0
forked from kscience/kmath
Code Pull Requests 1 Activity

Add withChecks to most tests

Browse Source
This commit is contained in:
Roland Grinis 2021-01-20 14:01:45 +00:00
parent 391eb28cad
commit c9dfb6a08c
4 changed files with 41 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kmath-torch/build.gradle.kts
View File

@ -187,4 +187,8 @@ tasks {
withType<Test>{
systemProperty("java.library.path", cppBuildDir.toString())
}
}
}
// No JS implementation
project.gradle.startParameter.excludedTaskNames.add("jsTest")
project.gradle.startParameter.excludedTaskNames.add("jsBrowserTest")

12
kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestAutograd.kt
View File

@ -6,15 +6,19 @@ import kotlin.test.*
internal class TestAutograd {
@Test
fun testAutoGrad() = TorchTensorFloatAlgebra {
withCuda { device ->
testingAutoGrad(device)
withChecks {
withCuda { device ->
testingAutoGrad(device)
}
}
}
@Test
fun testBatchedAutoGrad() = TorchTensorFloatAlgebra {
withCuda { device ->
testingBatchedAutoGrad(device)
withChecks {
withCuda { device ->
testingBatchedAutoGrad(device)
}
}
}
}

6
kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensor.kt
View File

@ -47,8 +47,10 @@ class TestTorchTensor {
@Test
fun testViewWithNoCopy() = TorchTensorIntAlgebra {
withCuda {
device -> testingViewWithNoCopy(device)
withChecks {
withCuda {
device -> testingViewWithNoCopy(device)
}
}
}

36
kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensorAlgebra.kt
View File

@ -7,43 +7,55 @@ internal class TestTorchTensorAlgebra {
@Test
fun testScalarProduct() = TorchTensorRealAlgebra {
withCuda { device ->
testingScalarProduct(device)
withChecks {
withCuda { device ->
testingScalarProduct(device)
}
}
}
@Test
fun testMatrixMultiplication() = TorchTensorRealAlgebra {
withCuda { device ->
testingMatrixMultiplication(device)
withChecks {
withCuda { device ->
testingMatrixMultiplication(device)
}
}
}
@Test
fun testLinearStructure() = TorchTensorRealAlgebra {
withCuda { device ->
testingLinearStructure(device)
withChecks {
withCuda { device ->
testingLinearStructure(device)
}
}
}
@Test
fun testTensorTransformations() = TorchTensorRealAlgebra {
withCuda { device ->
testingTensorTransformations(device)
withChecks {
withCuda { device ->
testingTensorTransformations(device)
}
}
}
@Test
fun testBatchedSVD() = TorchTensorRealAlgebra {
withCuda { device ->
testingBatchedSVD(device)
withChecks {
withCuda { device ->
testingBatchedSVD(device)
}
}
}
@Test
fun testBatchedSymEig() = TorchTensorRealAlgebra {
withCuda { device ->
testingBatchedSymEig(device)
withChecks {
withCuda { device ->
testingBatchedSymEig(device)
}
}
}