Skip to content

Commit

Permalink
clang-format for two files
Browse files Browse the repository at this point in the history
  • Loading branch information
qianfengz committed Aug 17, 2024
1 parent 2a5c141 commit 2874842
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,9 @@ struct batched_infer_causalmask_bias_dropout_dispatch {
// determine whether to do padding saving some compiling time
const bool pad_headdim = (pad_headdim_q || pad_headdim_v);

const bool use_async_pipeline =
(!kHasBias && (param.K % 8 == 0) && (param.Kv % 8 == 0) && (MaxK <= 128));
const bool use_async_pipeline =
(!kHasBias && (param.K % 8 == 0) && (param.Kv % 8 == 0) &&
(MaxK <= 128));

if (!use_async_pipeline) {
BOOL_SWITCH_3(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,8 @@ struct grouped_infer_causalmask_bias_dropout_dispatch {
bool pad_headdim_q = !(param.K % FmhaShape::kK0BlockLength == 0);
bool pad_headdim_v = !(param.Kv % FmhaShape::kN1 == 0);
const bool use_async_pipeline =
(!kHasBias && (param.K % 8 == 0) && (param.Kv % 8 == 0) && (MaxK <= 128));
(!kHasBias && (param.K % 8 == 0) && (param.Kv % 8 == 0) &&
(MaxK <= 128));

if (!use_async_pipeline) {
BOOL_SWITCH_2(
Expand Down

0 comments on commit 2874842

Please sign in to comment.