change ratio type to tensor

rusty1s · Nov 12, 2024 · 7ba8f3c · 7ba8f3c
1 parent 9d88a0b
commit 7ba8f3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/cluster.h b/csrc/cluster.h
@@ -10,7 +10,7 @@ CLUSTER_INLINE_VARIABLE int64_t _cuda_version = cuda_version();
 } // namespace detail
 } // namespace cluster
 
-CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, double ratio,
+CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, torch::Tensor ratio,
                   bool random_start);
 
 CLUSTER_API torch::Tensor graclus(torch::Tensor rowptr, torch::Tensor col,