This repo contains most of the ML primitives.
- cmake (>= 3.8 and <= 3.11.4, version 3.11.4 is recommended and there are some issues with version 3.12)
- CUDA SDK (>= 8.0)
- Cython (>= 0.28)
- gcc (>=5.4.0)
- nvcc (this comes with CUDA SDK)
ml-prims is implemented as header only C++/CUDA libraries for the developers who would like to call these APIs from their projects. You can build and run the Google tests if you are interested in helping us to improve these libraries.
First, clone the cuML if you haven't cloned it yet.
$ git clone --recursive firstname.lastname@example.org:rapidsai/cuml-alpha.git
To build ml-prims, in the main folder;
$ cd ml-prims $ mkdir build $ cd build $ cmake .. $ make -j $ ./mlcommon_test
The external folders inside ml-prims contain submodules that this project in-turn depends on. Appropriate location flags will be automatically populated in the main CMakeLists.txt file for these.
Current external submodules are: