Skip to content

Commit e53c87a

Browse files
Intron7Intron7
committed
add keywords
1 parent 84a34c4 commit e53c87a

38 files changed

Lines changed: 647 additions & 628 deletions

File tree

‎src/rapids_singlecell/_cuda/aucell/aucell.cu‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,6 @@ NB_MODULE(_aucell_cuda, m) {
5151
launch_auc(ranks, R, C, cnct, starts, lens, n_sets, n_up, max_aucs, es,
5252
(cudaStream_t)stream);
5353
},
54-
"ranks"_a, "R"_a, "C"_a, "cnct"_a, "starts"_a, "lens"_a, "n_sets"_a, "n_up"_a, "max_aucs"_a,
55-
"es"_a, "stream"_a = 0);
54+
"ranks"_a, nb::kw_only(), "R"_a, "C"_a, "cnct"_a, "starts"_a, "lens"_a, "n_sets"_a, "n_up"_a,
55+
"max_aucs"_a, "es"_a, "stream"_a = 0);
5656
}

‎src/rapids_singlecell/_cuda/autocorr/autocorr.cu‎

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ NB_MODULE(_autocorr_cuda, m) {
8080
launch_morans_dense(data_centered, adj_row_ptr, adj_col_ind, adj_data, num, n_samples,
8181
n_features, (cudaStream_t)stream);
8282
},
83-
"data_centered"_a, "adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "num"_a, "n_samples"_a,
84-
"n_features"_a, "stream"_a = 0);
83+
"data_centered"_a, nb::kw_only(), "adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "num"_a,
84+
"n_samples"_a, "n_features"_a, "stream"_a = 0);
8585
m.def(
8686
"morans_sparse",
8787
[](std::uintptr_t adj_row_ptr, std::uintptr_t adj_col_ind, std::uintptr_t adj_data,
@@ -92,8 +92,9 @@ NB_MODULE(_autocorr_cuda, m) {
9292
data_values, n_samples, n_features, mean_array, num,
9393
(cudaStream_t)stream);
9494
},
95-
"adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "data_row_ptr"_a, "data_col_ind"_a,
96-
"data_values"_a, "n_samples"_a, "n_features"_a, "mean_array"_a, "num"_a, "stream"_a = 0);
95+
"adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, nb::kw_only(), "data_row_ptr"_a,
96+
"data_col_ind"_a, "data_values"_a, "n_samples"_a, "n_features"_a, "mean_array"_a, "num"_a,
97+
"stream"_a = 0);
9798
m.def(
9899
"gearys_dense",
99100
[](std::uintptr_t data, std::uintptr_t adj_row_ptr, std::uintptr_t adj_col_ind,
@@ -102,8 +103,8 @@ NB_MODULE(_autocorr_cuda, m) {
102103
launch_gearys_dense(data, adj_row_ptr, adj_col_ind, adj_data, num, n_samples, n_features,
103104
(cudaStream_t)stream);
104105
},
105-
"data"_a, "adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "num"_a, "n_samples"_a,
106-
"n_features"_a, "stream"_a = 0);
106+
"data"_a, nb::kw_only(), "adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "num"_a,
107+
"n_samples"_a, "n_features"_a, "stream"_a = 0);
107108
m.def(
108109
"gearys_sparse",
109110
[](std::uintptr_t adj_row_ptr, std::uintptr_t adj_col_ind, std::uintptr_t adj_data,
@@ -112,8 +113,8 @@ NB_MODULE(_autocorr_cuda, m) {
112113
launch_gearys_sparse(adj_row_ptr, adj_col_ind, adj_data, data_row_ptr, data_col_ind,
113114
data_values, n_samples, n_features, num, (cudaStream_t)stream);
114115
},
115-
"adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, "data_row_ptr"_a, "data_col_ind"_a,
116-
"data_values"_a, "n_samples"_a, "n_features"_a, "num"_a, "stream"_a = 0);
116+
"adj_row_ptr"_a, "adj_col_ind"_a, "adj_data"_a, nb::kw_only(), "data_row_ptr"_a,
117+
"data_col_ind"_a, "data_values"_a, "n_samples"_a, "n_features"_a, "num"_a, "stream"_a = 0);
117118
m.def(
118119
"pre_den_sparse",
119120
[](std::uintptr_t data_col_ind, std::uintptr_t data_values, int nnz,
@@ -122,6 +123,6 @@ NB_MODULE(_autocorr_cuda, m) {
122123
launch_pre_den_sparse(data_col_ind, data_values, nnz, mean_array, den, counter,
123124
(cudaStream_t)stream);
124125
},
125-
"data_col_ind"_a, "data_values"_a, "nnz"_a, "mean_array"_a, "den"_a, "counter"_a,
126-
"stream"_a = 0);
126+
"data_col_ind"_a, "data_values"_a, nb::kw_only(), "nnz"_a, "mean_array"_a, "den"_a,
127+
"counter"_a, "stream"_a = 0);
127128
}

‎src/rapids_singlecell/_cuda/bbknn/bbknn.cu‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,13 @@ NB_MODULE(_bbknn_cuda, m) {
4040
std::uintptr_t stream) {
4141
launch_find_top_k_per_row(data, indptr, n_rows, trim, vals, (cudaStream_t)stream);
4242
},
43-
"data"_a, "indptr"_a, "n_rows"_a, "trim"_a, "vals"_a, "stream"_a = 0);
43+
"data"_a, "indptr"_a, nb::kw_only(), "n_rows"_a, "trim"_a, "vals"_a, "stream"_a = 0);
4444

4545
m.def(
4646
"cut_smaller",
4747
[](std::uintptr_t indptr, std::uintptr_t index, std::uintptr_t data, std::uintptr_t vals,
4848
int n_rows, std::uintptr_t stream) {
4949
launch_cut_smaller(indptr, index, data, vals, n_rows, (cudaStream_t)stream);
5050
},
51-
"indptr"_a, "index"_a, "data"_a, "vals"_a, "n_rows"_a, "stream"_a = 0);
51+
"indptr"_a, "index"_a, "data"_a, nb::kw_only(), "vals"_a, "n_rows"_a, "stream"_a = 0);
5252
}

‎src/rapids_singlecell/_cuda/cooc/cooc.cu‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -91,21 +91,23 @@ NB_MODULE(_cooc_cuda, m) {
9191
launch_count_pairwise(spatial, thresholds, labels, result, n, k, l_val,
9292
(cudaStream_t)stream);
9393
},
94-
"spatial"_a, "thresholds"_a, "labels"_a, "result"_a, "n"_a, "k"_a, "l_val"_a, "stream"_a = 0);
94+
"spatial"_a, nb::kw_only(), "thresholds"_a, "labels"_a, "result"_a, "n"_a, "k"_a, "l_val"_a,
95+
"stream"_a = 0);
9596
m.def(
9697
"reduce_shared",
9798
[](std::uintptr_t result, std::uintptr_t out, int k, int l_val, int format,
9899
std::uintptr_t stream) {
99100
return launch_reduce_shared(result, out, k, l_val, format, (cudaStream_t)stream);
100101
},
101-
"result"_a, "out"_a, "k"_a, "l_val"_a, "format"_a, "stream"_a = 0);
102+
"result"_a, nb::kw_only(), "out"_a, "k"_a, "l_val"_a, "format"_a, "stream"_a = 0);
102103
m.def(
103104
"reduce_global",
104105
[](std::uintptr_t result, std::uintptr_t inter_out, std::uintptr_t out, int k, int l_val,
105106
int format, std::uintptr_t stream) {
106107
launch_reduce_global(result, inter_out, out, k, l_val, format, (cudaStream_t)stream);
107108
},
108-
"result"_a, "inter_out"_a, "out"_a, "k"_a, "l_val"_a, "format"_a, "stream"_a = 0);
109+
"result"_a, nb::kw_only(), "inter_out"_a, "out"_a, "k"_a, "l_val"_a, "format"_a,
110+
"stream"_a = 0);
109111
m.def(
110112
"count_csr_catpairs_auto",
111113
[](std::uintptr_t spatial, std::uintptr_t thresholds, std::uintptr_t cat_offsets,
@@ -115,6 +117,6 @@ NB_MODULE(_cooc_cuda, m) {
115117
pair_left, pair_right, counts_delta, num_pairs, k,
116118
l_val, (cudaStream_t)stream);
117119
},
118-
"spatial"_a, "thresholds"_a, "cat_offsets"_a, "cell_indices"_a, "pair_left"_a, "pair_right"_a,
119-
"counts_delta"_a, "num_pairs"_a, "k"_a, "l_val"_a, "stream"_a = 0);
120+
"spatial"_a, nb::kw_only(), "thresholds"_a, "cat_offsets"_a, "cell_indices"_a, "pair_left"_a,
121+
"pair_right"_a, "counts_delta"_a, "num_pairs"_a, "k"_a, "l_val"_a, "stream"_a = 0);
120122
}

‎src/rapids_singlecell/_cuda/harmony/colsum/colsum.cu‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ NB_MODULE(_harmony_colsum_cuda, m) {
4343
throw nb::value_error("Unsupported dtype_code (expected 0/1/2 or 4/8)");
4444
}
4545
},
46-
"A"_a, "out"_a, "rows"_a, "cols"_a, "dtype_code"_a, "stream"_a = 0);
46+
"A"_a, nb::kw_only(), "out"_a, "rows"_a, "cols"_a, "dtype_code"_a, "stream"_a = 0);
4747

4848
m.def(
4949
"colsum_atomic",
@@ -59,5 +59,5 @@ NB_MODULE(_harmony_colsum_cuda, m) {
5959
throw nb::value_error("Unsupported dtype_code (expected 0/1/2 or 4/8)");
6060
}
6161
},
62-
"A"_a, "out"_a, "rows"_a, "cols"_a, "dtype_code"_a, "stream"_a = 0);
62+
"A"_a, nb::kw_only(), "out"_a, "rows"_a, "cols"_a, "dtype_code"_a, "stream"_a = 0);
6363
}

‎src/rapids_singlecell/_cuda/harmony/kmeans/kmeans.cu‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,5 +29,5 @@ NB_MODULE(_harmony_kmeans_cuda, m) {
2929
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
3030
}
3131
},
32-
"r"_a, "dot"_a, "n"_a, "out"_a, "itemsize"_a, "stream"_a = 0);
32+
"r"_a, nb::kw_only(), "dot"_a, "n"_a, "out"_a, "itemsize"_a, "stream"_a = 0);
3333
}

‎src/rapids_singlecell/_cuda/harmony/normalize/normalize.cu‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,5 @@ NB_MODULE(_harmony_normalize_cuda, m) {
2727
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
2828
}
2929
},
30-
"X"_a, "rows"_a, "cols"_a, "itemsize"_a, "stream"_a = 0);
30+
"X"_a, nb::kw_only(), "rows"_a, "cols"_a, "itemsize"_a, "stream"_a = 0);
3131
}

‎src/rapids_singlecell/_cuda/harmony/outer/outer.cu‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ NB_MODULE(_harmony_outer_cuda, m) {
4444
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
4545
}
4646
},
47-
"E"_a, "Pr_b"_a, "R_sum"_a, "n_cats"_a, "n_pcs"_a, "switcher"_a, "itemsize"_a,
47+
"E"_a, nb::kw_only(), "Pr_b"_a, "R_sum"_a, "n_cats"_a, "n_pcs"_a, "switcher"_a, "itemsize"_a,
4848
"stream"_a = 0);
4949

5050
m.def(
@@ -59,5 +59,6 @@ NB_MODULE(_harmony_outer_cuda, m) {
5959
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
6060
}
6161
},
62-
"Z"_a, "W"_a, "cats"_a, "R"_a, "n_cells"_a, "n_pcs"_a, "itemsize"_a, "stream"_a = 0);
62+
"Z"_a, nb::kw_only(), "W"_a, "cats"_a, "R"_a, "n_cells"_a, "n_pcs"_a, "itemsize"_a,
63+
"stream"_a = 0);
6364
}

‎src/rapids_singlecell/_cuda/harmony/pen/pen.cu‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,5 +31,6 @@ NB_MODULE(_harmony_pen_cuda, m) {
3131
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
3232
}
3333
},
34-
"R"_a, "penalty"_a, "cats"_a, "n_rows"_a, "n_cols"_a, "itemsize"_a, "stream"_a = 0);
34+
"R"_a, nb::kw_only(), "penalty"_a, "cats"_a, "n_rows"_a, "n_cols"_a, "itemsize"_a,
35+
"stream"_a = 0);
3536
}

‎src/rapids_singlecell/_cuda/harmony/scatter/scatter.cu‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,8 @@ NB_MODULE(_harmony_scatter_cuda, m) {
6666
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
6767
}
6868
},
69-
"v"_a, "cats"_a, "n_cells"_a, "n_pcs"_a, "switcher"_a, "a"_a, "itemsize"_a, "stream"_a = 0);
69+
"v"_a, nb::kw_only(), "cats"_a, "n_cells"_a, "n_pcs"_a, "switcher"_a, "a"_a, "itemsize"_a,
70+
"stream"_a = 0);
7071

7172
m.def(
7273
"aggregated_matrix",
@@ -82,7 +83,8 @@ NB_MODULE(_harmony_scatter_cuda, m) {
8283
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
8384
}
8485
},
85-
"aggregated_matrix"_a, "sum"_a, "top_corner"_a, "n_batches"_a, "itemsize"_a, "stream"_a = 0);
86+
"aggregated_matrix"_a, nb::kw_only(), "sum"_a, "top_corner"_a, "n_batches"_a, "itemsize"_a,
87+
"stream"_a = 0);
8688

8789
m.def(
8890
"scatter_add_cat0",
@@ -96,7 +98,7 @@ NB_MODULE(_harmony_scatter_cuda, m) {
9698
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
9799
}
98100
},
99-
"v"_a, "n_cells"_a, "n_pcs"_a, "a"_a, "bias"_a, "itemsize"_a, "stream"_a = 0);
101+
"v"_a, nb::kw_only(), "n_cells"_a, "n_pcs"_a, "a"_a, "bias"_a, "itemsize"_a, "stream"_a = 0);
100102

101103
m.def(
102104
"scatter_add_block",
@@ -113,6 +115,6 @@ NB_MODULE(_harmony_scatter_cuda, m) {
113115
throw nb::value_error("Unsupported itemsize (expected 4 or 8)");
114116
}
115117
},
116-
"v"_a, "cat_offsets"_a, "cell_indices"_a, "n_cells"_a, "n_pcs"_a, "n_batches"_a, "a"_a,
117-
"bias"_a, "itemsize"_a, "stream"_a = 0);
118+
"v"_a, nb::kw_only(), "cat_offsets"_a, "cell_indices"_a, "n_cells"_a, "n_pcs"_a,
119+
"n_batches"_a, "a"_a, "bias"_a, "itemsize"_a, "stream"_a = 0);
118120
}

0 commit comments

Comments
 (0)