Skip to content
/ dpu_kmeans Public template
generated from SylvanBrocard/dpu_trees

Implementation of the K-means algorithm on UPMEM PIM architecture

License

Notifications You must be signed in to change notification settings

upmem/dpu_kmeans

Repository files navigation

dpu_trees

CI status
pip builds Pip Actions Status

A project built with pybind11 and scikit-build, running DPU programs with the UPMEM SDK.

Installation

  • install the UPMEM SDK
  • pip install dpu-trees

OR

  • clone this repository
  • install the UPMEM SDK
  • pip install ./dpu_trees

Development

  • clone this repository
  • install the UPMEM SDK
  • cd dpu_trees
  • pre-commit install
  • python3 setup.py develop
  • python setup.py clean

OR

  • clone this repository
  • open folder in VS Code
  • start in Dev Container

Testing

  • clone this repository
  • install the UPMEM SDK
  • install nox
  • cd dpu_trees
  • nox

OR

  • clone this repository
  • open folder in VS Code
  • start in Dev Container
  • nox

Note: nox and pip will fail if you executed setup.py install or setup.py develop, delete the cache by running python setup.py clean to solve.

Test call

import dpu_trees
dpu_trees.add(1, 2)

Expected return: 0x007f8000

About

Implementation of the K-means algorithm on UPMEM PIM architecture

Resources

License

Stars

Watchers

Forks

Packages

No packages published