Skip to content

Commit 84e1ef1

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c87565e commit 84e1ef1

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

transformer_engine/pytorch/ops/fused/backward_grouped_mlp.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -330,8 +330,7 @@ def __init__(
330330
# cuDNN-frontend >= 1.24.0 exposes runtime-configurable GeGLU
331331
# parameters; pass them through when available.
332332
self._pass_geglu_runtime_params: bool = (
333-
isinstance(activation, ScaledClampedQGeGLU)
334-
and _cudnn_frontend_geglu_runtime_params()
333+
isinstance(activation, ScaledClampedQGeGLU) and _cudnn_frontend_geglu_runtime_params()
335334
)
336335
if self._pass_geglu_runtime_params:
337336
self._cudnn_linear_offset: float = activation._clamped.glu_linear_offset

transformer_engine/pytorch/ops/fused/forward_grouped_mlp.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,7 @@ def __init__(
133133
# parameters; pass them through when the activation carries
134134
# non-default values (or always, if available).
135135
self._pass_geglu_runtime_params: bool = (
136-
isinstance(activation, ScaledClampedQGeGLU)
137-
and _cudnn_frontend_geglu_runtime_params()
136+
isinstance(activation, ScaledClampedQGeGLU) and _cudnn_frontend_geglu_runtime_params()
138137
)
139138
if self._pass_geglu_runtime_params:
140139
self._cudnn_linear_offset: float = activation._clamped.glu_linear_offset

0 commit comments

Comments
 (0)