drbh
commited on
Commit
·
6c3907a
1
Parent(s):
3099d65
fix: adjust comma typos
Browse files
torch-ext/torch_binding.h
CHANGED
@@ -10,11 +10,11 @@ mha_fwd(torch::Tensor &q,
|
|
10 |
c10::optional<torch::Tensor> alibi_slopes_,
|
11 |
const double p_dropout,
|
12 |
const double softmax_scale,
|
13 |
-
bool is_causal
|
14 |
-
const int64_t window_size_left
|
15 |
const int64_t window_size_right,
|
16 |
const double softcap,
|
17 |
-
const bool return_softmax
|
18 |
c10::optional<at::Generator> gen_);
|
19 |
|
20 |
std::vector<torch::Tensor>
|
|
|
10 |
c10::optional<torch::Tensor> alibi_slopes_,
|
11 |
const double p_dropout,
|
12 |
const double softmax_scale,
|
13 |
+
bool is_causal,
|
14 |
+
const int64_t window_size_left,
|
15 |
const int64_t window_size_right,
|
16 |
const double softcap,
|
17 |
+
const bool return_softmax,
|
18 |
c10::optional<at::Generator> gen_);
|
19 |
|
20 |
std::vector<torch::Tensor>
|