Skip to content

Commit e1d99de

Browse files
change ratio type to tensor (#234)
1 parent 9d88a0b commit e1d99de

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/cluster.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ CLUSTER_INLINE_VARIABLE int64_t _cuda_version = cuda_version();
1010
} // namespace detail
1111
} // namespace cluster
1212

13-
CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, double ratio,
13+
CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, torch::Tensor ratio,
1414
bool random_start);
1515

1616
CLUSTER_API torch::Tensor graclus(torch::Tensor rowptr, torch::Tensor col,

0 commit comments

Comments
 (0)