File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff line change 2121from thunder .executors .sdpaex import sdpa_ex
2222from 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.
2627def _detect_has_args_tensor_mask ():
You can’t perform that action at this time.
0 commit comments