Add withChecks to most tests

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

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")

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)
}
}
}
}

View File

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

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)
}
}
}