Package de.jungblut.math.cuda

Source Code of de.jungblut.math.cuda.MatrixDimensionTest

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package de.jungblut.math.cuda;

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import de.jungblut.math.dense.DenseDoubleMatrix;

public class MatrixDimensionTest {

  @Test
  public void testDimensionalityNoTranspose() {

    MatrixDimension dimension = new MatrixDimension(
        new DenseDoubleMatrix(2, 3), new DenseDoubleMatrix(15, 2));

    assertEquals(2, dimension.getM());
    assertEquals(2, dimension.getN());
    assertEquals(3, dimension.getK());

    assertEquals(2, dimension.getLdA());
    assertEquals(15, dimension.getLdB());
    assertEquals(2, dimension.getLdC());

    assertEquals(false, dimension.isTransposeA());
    assertEquals(false, dimension.isTransposeB());

  }

  @Test
  public void testDimensionality() {

    MatrixDimension dimension = new MatrixDimension(
        new DenseDoubleMatrix(3, 2), new DenseDoubleMatrix(15, 2), true, false);

    assertEquals(2, dimension.getM());
    assertEquals(2, dimension.getN());
    assertEquals(3, dimension.getK());

    assertEquals(3, dimension.getLdA());
    assertEquals(15, dimension.getLdB());
    assertEquals(2, dimension.getLdC());

    assertEquals(true, dimension.isTransposeA());
    assertEquals(false, dimension.isTransposeB());

    dimension = new MatrixDimension(new DenseDoubleMatrix(3, 2),
        new DenseDoubleMatrix(2, 15), true, true);
    assertEquals(2, dimension.getM());
    assertEquals(2, dimension.getN());
    assertEquals(15, dimension.getK());

    assertEquals(3, dimension.getLdA());
    assertEquals(2, dimension.getLdB());
    assertEquals(2, dimension.getLdC());

    assertEquals(true, dimension.isTransposeA());
    assertEquals(true, dimension.isTransposeB());

    assertEquals(
        "MatrixDimension [m=2, n=2, k=15, ldA=3, ldB=2, ldC=2, transposeA=true, transposeB=true]",
        dimension.toString());

    dimension = new MatrixDimension(new DenseDoubleMatrix(3, 2),
        new DenseDoubleMatrix(2, 15), false, true);
    assertEquals(3, dimension.getM());
    assertEquals(2, dimension.getN());
    assertEquals(2, dimension.getK());

    assertEquals(3, dimension.getLdA());
    assertEquals(2, dimension.getLdB());
    assertEquals(3, dimension.getLdC());

    assertEquals(false, dimension.isTransposeA());
    assertEquals(true, dimension.isTransposeB());

  }

}
TOP

Related Classes of de.jungblut.math.cuda.MatrixDimensionTest

  • de.jungblut.math.dense.DenseDoubleMatrix
    • TOP
    Copyright © 2018 www.massapi.com. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.