diff --git a/kmath-torch/build.gradle.kts b/kmath-torch/build.gradle.kts index 1508f0ef7..406f3a2a5 100644 --- a/kmath-torch/build.gradle.kts +++ b/kmath-torch/build.gradle.kts @@ -187,4 +187,8 @@ tasks { withType{ systemProperty("java.library.path", cppBuildDir.toString()) } -} \ No newline at end of file +} + +// No JS implementation +project.gradle.startParameter.excludedTaskNames.add("jsTest") +project.gradle.startParameter.excludedTaskNames.add("jsBrowserTest") \ No newline at end of file diff --git a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestAutograd.kt b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestAutograd.kt index 6560153f7..3939d5e93 100644 --- a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestAutograd.kt +++ b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestAutograd.kt @@ -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) + } } } } \ No newline at end of file diff --git a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensor.kt b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensor.kt index 4a8edfd58..951294ecb 100644 --- a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensor.kt +++ b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensor.kt @@ -47,8 +47,10 @@ class TestTorchTensor { @Test fun testViewWithNoCopy() = TorchTensorIntAlgebra { - withCuda { - device -> testingViewWithNoCopy(device) + withChecks { + withCuda { + device -> testingViewWithNoCopy(device) + } } } diff --git a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensorAlgebra.kt b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensorAlgebra.kt index c761ab1ca..d05de3843 100644 --- a/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensorAlgebra.kt +++ b/kmath-torch/src/nativeTest/kotlin/kscience/kmath/torch/TestTorchTensorAlgebra.kt @@ -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) + } } }