You need to sign in or sign up before continuing.
Commit 3c64bb0d authored by rusty1s's avatar rusty1s
Browse files

update readme [ci skip]

parent 723039b8
...@@ -294,21 +294,22 @@ print(valueC) ...@@ -294,21 +294,22 @@ print(valueC)
tensor([8.0, 6.0, 8.0]) tensor([8.0, 6.0, 8.0])
``` ```
## Running tests
```
pytest
```
## C++ API ## C++ API
`torch-sparse` also offers a C++ API that contains C++ equivalent of python models. `torch-sparse` also offers a C++ API that contains C++ equivalent of python models.
For this, we need to add `TorchLib` to the `-DCMAKE_PREFIX_PATH` (*e.g.*, it may exists in `{CONDA}/lib/python{X.X}/site-packages/torch` if installed via `conda`):
``` ```
mkdir build mkdir build
cd build cd build
# Add -DWITH_CUDA=on support for the CUDA if needed # Add -DWITH_CUDA=on support for CUDA support
cmake .. cmake -DCMAKE_PREFIX_PATH="..." ..
make make
make install make install
``` ```
## Running tests
```
pytest
```
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment