Skip to content

Commit d734891

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

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/pytorch/attention/run_attention_with_cp_pool.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def _reset_between_cases() -> None:
9292
# instead of reusing the previous case's resolved backend.
9393
try:
9494
from transformer_engine.pytorch.attention.dot_product_attention import dot_product_attention
95+
9596
dot_product_attention._attention_backends["backend_selection_requires_update"] = True
9697
except (ImportError, AttributeError, KeyError):
9798
pass

0 commit comments

Comments
 (0)