コード例 #1
0
ファイル: test_pdfs.py プロジェクト: r0k3/gpustats 
def _compare_multi(n, k, p):
    data, means, covs = _make_test_case(n, k, p)

    # cpu in PyMC
    pyresult = compat.python_mvnpdf(data, means, covs)

    # gpu
    result = gps.mvnpdf_multi(data, means, covs)

    return result, pyresult
コード例 #2
0
ファイル: test_pdfs.py プロジェクト: jfrelinger/gpustats 
def _compare_multi(n, k, p):
    data, means, covs = _make_test_case(n, k, p)

    # cpu in PyMC
    pyresult = compat.python_mvnpdf(data, means, covs)

    # gpu
    result = gps.mvnpdf_multi(data, means, covs)

    return result, pyresult
コード例 #3
0
ファイル: test_pdfs.py プロジェクト: r0k3/gpustats 
def _compare_single(n, k):
    data, means, covs = _make_test_case(n, k, 1)

    mean = means[0]
    cov = covs[0]

    # cpu in PyMC
    pyresult = compat.python_mvnpdf(data, [mean], [cov]).squeeze()
    # gpu

    result = gps.mvnpdf(data, mean, cov)
    return result, pyresult
コード例 #4
0
ファイル: test_pdfs.py プロジェクト: jfrelinger/gpustats 
def _compare_single(n, k):
    data, means, covs = _make_test_case(n, k, 1)

    mean = means[0]
    cov = covs[0]

    # cpu in PyMC
    pyresult = compat.python_mvnpdf(data, [mean], [cov]).squeeze()
    # gpu

    result = gps.mvnpdf(data, mean, cov)
    return result, pyresult