kernel
drbh commited on
Commit
6c3907a
·
1 Parent(s): 3099d65

fix: adjust comma typos

Browse files
Files changed (1) hide show
  1. torch-ext/torch_binding.h +3 -3
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>