More test tagging

master
agibsonccc 2021-03-26 11:15:26 +09:00
parent 6e7a843d2d
commit 4f4939534a
6 changed files with 13 additions and 1 deletions

View File

@ -22,6 +22,6 @@ jobs:
cmake --version
protoc --version
export OMP_NUM_THREADS=1
mvn -DexcludedGroups="long-running-tests,large-resources, distributed-systems" -DskipTestResourceEnforcement=true -Ptestresources -Pintegration-tests -Pnd4j-tests-cpu clean test
mvn -DexcludedGroups="long-running-tests,large-resources, distributed-systems" -DskipTestResourceEnforcement=true -Ptestresources -Pintegration-tests -Pnd4j-tests-cpu clean test -rf :deeplearning4j-core
mvn -Dgroups="long-running-tests, large-resources, distributed-systems" -Ptestresources -Pnd4j-tests-cpu -Dtest.offheap.size=14g -Dtest.heap.size=6g -Dsurefire.parallel.forcedTimeout=200 0Dsurefire.parallel.timeout=200 -Dsurefire.timeout=200 -Dsurefire.exitTimeout=200 test

View File

@ -327,6 +327,8 @@ public class TestComputationGraphNetwork extends BaseDL4JTest {
@Test()
@Timeout(300000)
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
public void testIrisFitMultiDataSetIterator() throws Exception {
RecordReader rr = new CSVRecordReader(0, ',');

View File

@ -81,6 +81,8 @@ class DropoutLayerTest extends BaseDL4JTest {
@Test
@DisplayName("Test Dropout Layer Without Training")
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
void testDropoutLayerWithoutTraining() throws Exception {
MultiLayerConfiguration confIntegrated = new NeuralNetConfiguration.Builder().seed(3648).list().layer(0, new ConvolutionLayer.Builder(1, 1).stride(1, 1).nIn(1).nOut(1).dropOut(0.25).activation(Activation.IDENTITY).weightInit(WeightInit.XAVIER).build()).layer(1, new OutputLayer.Builder(LossFunctions.LossFunction.MCXENT).activation(Activation.SOFTMAX).weightInit(WeightInit.XAVIER).dropOut(0.25).nOut(4).build()).setInputType(InputType.convolutionalFlat(2, 2, 1)).build();
MultiLayerNetwork netIntegrated = new MultiLayerNetwork(confIntegrated);

View File

@ -65,6 +65,8 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
@AllArgsConstructor
@NativeTag
@Tag(TagNames.DL4J_OLD_API)
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.LARGE_RESOURCES)
public class RnnDataFormatTests extends BaseDL4JTest {

View File

@ -153,6 +153,8 @@ class LegacyWeightInitTest extends BaseDL4JTest {
*/
@Test
@DisplayName("Serialize Deserialize Distribution Json")
@Disabled("")
@Tag(TagNames.NEEDS_VERIFY)
void serializeDeserializeDistributionJson() throws IOException {
// To make identity happy
final long[] shape = { 3, 7 };

View File

@ -39,6 +39,7 @@ import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test;
import org.nd4j.common.tests.tags.NativeTag;
import org.nd4j.common.tests.tags.TagNames;
import org.nd4j.linalg.activations.Activation;
import org.nd4j.linalg.api.buffer.DataType;
import org.nd4j.linalg.api.ndarray.INDArray;
@ -58,6 +59,8 @@ import org.junit.jupiter.api.extension.ExtendWith;
*/
@NativeTag
@DisplayName("Cnn Gradient Check Test")
@Tag(TagNames.LARGE_RESOURCES)
@Tag(TagNames.LONG_TEST)
class CNNGradientCheckTest extends BaseDL4JTest {
private static final boolean PRINT_RESULTS = true;
@ -81,6 +84,7 @@ class CNNGradientCheckTest extends BaseDL4JTest {
@Test
@DisplayName("Test Gradient CNNMLN")
void testGradientCNNMLN() {
// Parameterized test, testing combinations of:
// (a) activation function