Skip to content

[BUG] fuzzy_simplicial_set() returns different outputs on CPU/GPU #6344

Open
@willleeney

Description

@willleeney

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions