Esempio n. 1
0
def test_ddot():

    a = np.array([0, 1, 2, 3], 'f8')
    b = np.array([0, 2, 3, 4], 'f8')
    assert_equal(ddot(a, b), np.dot(a, b))
Esempio n. 2
0
make PREFIX=. install
Check for include and library directories
"""


if __name__ == '__main__':

    """
    Level 1 example
    """

    a = np.array([0, 1, 2, 3], 'f8')
    b = np.array([0, 2, 3, 4], 'f8')
    print(a)
    print(b)
    print(ddot(a, b))
    
    """
    Level 3
    alpha*dot(A,B) + beta*C
    """

    A = np.array([[0, 1, 2], [3, 4, 5]], dtype='f8')
    B = np.array([[0, 1], [2, 3], [4, 5]], dtype='f8')
    C = np.zeros((2, 2))
    alpha = 2.
    beta = 3.


    dgemm(A, B, C, alpha, beta)
    print(C)