Open
Description
Describe the bug
fuzzy_simplicial_set() returns different output on CPU and GPU
Steps/Code to reproduce bug
run on GPU and CPU:
import numpy as np
from cuml.manifold.umap import fuzzy_simplicial_set
X = np.random.rand(1000, 50).astype(np.float32)
graph = fuzzy_simplicial_set(X, n_neighbors=15, metric='euclidean')
on CPU it returns a tuple containing a csr_matrix
on GPU it returns a coo_matrix
Expected behavior
I would've thought that on the CPU it would also return a coo_matrix
Environment details (please complete the following information):
- Environment location: Docker, Google Collab
- Linux Distro/Architecture: Ubuntu 16.04 amd64
- GPU Model/Driver: T4
- cuml version: 24.12