Python pwdist_cl示例

示例#1

文件： opencl-pairwise.py 项目： mertdikmen/ViVid

import pathfinder
import vivid
import numpy as np

A = np.random.random((5,1000)).astype('float32')
B = np.random.random((5,1000)).astype('float32')

dA = vivid.DeviceMatrixCL(A)
dB = vivid.DeviceMatrixCL(B)

dC = vivid.pwdist_cl(dA,dB)
C = dC.mat()

示例#2

文件： pairwise_distance.py 项目： mertdikmen/ViVid

# Pairwise distance using Python (for reference)
reference = A[:,np.newaxis,:] - B[np.newaxis,:,:]
Cpython = (reference * reference).sum(axis=2)

DO_OPENCL = False
DO_CUDA = True

if DO_OPENCL:
    dAcl = vivid.DeviceMatrixCL(A)
    dBcl = vivid.DeviceMatrixCL(B)

    # Pairwise distance using OpenCL
    tic = time.time()
    for i in range(10):
        Ccl = vivid.pwdist_cl(dAcl, dBcl).mat()
    toc = time.time()
    print("CL Time (Python): {}".format(toc - tic))

    if np.allclose(Ccl, Cpython, 1e-4):
        print("OpenCL Result OK")

if DO_CUDA:
    dA = vivid.DeviceMatrix(A)
    dB = vivid.DeviceMatrix(B)

    # Pairwise distance using CUDA
    tic = time.time()
    for i in range(10):
        Ccuda = vivid.pwdist_cuda(dA,dB).mat()
    toc = time.time()