Skip to content

a simple k-nearest-neighbour implemented with pyopencl

Notifications You must be signed in to change notification settings

Jackie-Chou/pyopencl_knn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

pyopencl_knn

a simple k-nearest-neighbour implemented with pyopencl

file description

check: folder containing opencl code to check the configuration of the machine and opencl

src: folder containing source code

  • knn: main code file, configure the host and run opencl. NOTE the platform choice is hard coded to 0, one may feel free to change it. run python knn.py -h for arguments and usage.also NOTE this file reads input feature map and kernel from input .npz file, and result will be stored to an output .npz file, example files please refer to input.npz and output.npz
  • kernels.cl: opencl kernel function file
  • check.py: a simple correctness checking script by compare the result with a naive serial algorithm.

About

a simple k-nearest-neighbour implemented with pyopencl

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published