Skip to content

Commit d03f03d

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

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

thunder/core/jit_ext.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -970,7 +970,7 @@ def _generate_random_str_id() -> str:
970970
# With args_tensor_mask, the fwd_body expects ctx as first argument
971971
new_fwd_args = (wrap_const(None),) + tuple(new_fwd_args)
972972
else:
973-
# For nightly PyTorch without args_tensor_mask, the fwd_body
973+
# For nightly PyTorch without args_tensor_mask, the fwd_body
974974
# GraphModule does NOT expect a ctx argument.
975975
# We pass all args as-is without prepending None.
976976
new_fwd_args = tuple(fwd_args)

thunder/tests/test_jit_general.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from thunder.executors.sdpaex import sdpa_ex
2222
from thunder.core.transforms import Transform
2323

24+
2425
# Detect once at module load time whether PyTorch uses args_tensor_mask.
2526
# This must be done outside the JIT-traced function to avoid interpreter issues.
2627
def _detect_has_args_tensor_mask():

0 commit comments

Comments
 (0)