Skip to content

Commit 63f7fa6

Browse files
committed
refactor
1 parent 8d8a904 commit 63f7fa6

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

python/cuml/cuml/manifold/spectral_embedding.pyx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def spectral_embedding(A,
171171
input_to_cuml_array(A, order="C", check_dtype=np.float32,
172172
convert_to_dtype=cp.float32)
173173
A_ptr = <uintptr_t>A.ptr
174-
n_samples = A.shape[0]
174+
n_samples = _n_rows
175175

176176
cdef params config
177177

@@ -206,9 +206,9 @@ def spectral_embedding(A,
206206
transform(
207207
deref(h), config,
208208
make_device_matrix_view[float, int, row_major](
209-
<float *>A_ptr, <int> n_samples, <int> A.shape[1]),
209+
<float *>A_ptr, <int> _n_rows, <int> _n_cols),
210210
make_device_matrix_view[float, int, col_major](
211-
<float *>eigenvectors_ptr, <int> n_samples, <int> n_components))
211+
<float *>eigenvectors_ptr, <int> _n_rows, <int> n_components))
212212

213213
return eigenvectors
214214

0 commit comments

Comments
 (0)