235
235
< div class ="pytorch-left-menu-search ">
236
236
237
237
< div class ="version ">
238
- < a href ='https://pytorch.org/docs/versions.html '> master (2.1.0a0+git9a2a6fc ) ▼</ a >
238
+ < a href ='https://pytorch.org/docs/versions.html '> master (2.1.0a0+gitfd7eaf7 ) ▼</ a >
239
239
</ div >
240
240
241
241
@@ -838,7 +838,9 @@ <h1>Source code for torch._dynamo.eval_frame</h1><div class="highlight"><pre>
838
838
839
839
< span class ="k "> def</ span > < span class ="nf "> catch_errors_wrapper</ span > < span class ="p "> (</ span > < span class ="n "> callback</ span > < span class ="p "> ,</ span > < span class ="n "> hooks</ span > < span class ="p "> :</ span > < span class ="n "> Hooks</ span > < span class ="p "> ):</ span >
840
840
< span class ="nd "> @functools</ span > < span class ="o "> .</ span > < span class ="n "> wraps</ span > < span class ="p "> (</ span > < span class ="n "> callback</ span > < span class ="p "> )</ span >
841
- < span class ="k "> def</ span > < span class ="nf "> catch_errors</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ):</ span >
841
+ < span class ="k "> def</ span > < span class ="nf "> catch_errors</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ,</ span > < span class ="n "> frame_state</ span > < span class ="p "> ):</ span >
842
+ < span class ="k "> assert</ span > < span class ="n "> frame_state</ span > < span class ="ow "> is</ span > < span class ="ow "> not</ span > < span class ="kc "> None</ span >
843
+
842
844
< span class ="k "> if</ span > < span class ="p "> (</ span >
843
845
< span class ="c1 "> # TODO: the first condition is not covered by any test</ span >
844
846
< span class ="n "> frame</ span > < span class ="o "> .</ span > < span class ="n "> f_lasti</ span > < span class ="o "> >=</ span > < span class ="n "> first_real_inst_idx</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="o "> .</ span > < span class ="n "> f_code</ span > < span class ="p "> )</ span >
@@ -864,10 +866,10 @@ <h1>Source code for torch._dynamo.eval_frame</h1><div class="highlight"><pre>
864
866
< span class ="n "> ddp_optimizer</ span > < span class ="o "> .</ span > < span class ="n "> compile_fn</ span > < span class ="p "> ,</ span >
865
867
< span class ="n "> hooks</ span > < span class ="o "> =</ span > < span class ="n "> hooks</ span > < span class ="p "> ,</ span >
866
868
< span class ="p "> )</ span >
867
- < span class ="k "> return</ span > < span class ="n "> hijacked_callback</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ,</ span > < span class ="n "> hooks</ span > < span class ="p "> )</ span >
869
+ < span class ="k "> return</ span > < span class ="n "> hijacked_callback</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ,</ span > < span class ="n "> hooks</ span > < span class ="p "> , </ span > < span class =" n " > frame_state </ span > < span class =" p " > )</ span >
868
870
869
871
< span class ="k "> with</ span > < span class ="n "> compile_lock</ span > < span class ="p "> :</ span >
870
- < span class ="k "> return</ span > < span class ="n "> callback</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ,</ span > < span class ="n "> hooks</ span > < span class ="p "> )</ span >
872
+ < span class ="k "> return</ span > < span class ="n "> callback</ span > < span class ="p "> (</ span > < span class ="n "> frame</ span > < span class ="p "> ,</ span > < span class ="n "> cache_size</ span > < span class ="p "> ,</ span > < span class ="n "> hooks</ span > < span class ="p "> , </ span > < span class =" n " > frame_state </ span > < span class =" p " > )</ span >
871
873
872
874
< span class ="n "> catch_errors</ span > < span class ="o "> .</ span > < span class ="n "> _torchdynamo_orig_callable</ span > < span class ="o "> =</ span > < span class ="n "> callback</ span > < span class ="c1 "> # type: ignore[attr-defined]</ span >
873
875
< span class ="k "> return</ span > < span class ="n "> catch_errors</ span >
@@ -1082,7 +1084,8 @@ <h1>Source code for torch._dynamo.eval_frame</h1><div class="highlight"><pre>
1082
1084
1083
1085
< span class ="k "> def</ span > < span class ="nf "> _clone_with_range</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> lower</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ,</ span > < span class ="n "> upper</ span > < span class ="o "> =</ span > < span class ="n "> sympy</ span > < span class ="o "> .</ span > < span class ="n "> oo</ span > < span class ="p "> ):</ span >
1084
1086
< span class ="n "> constraint_range</ span > < span class ="o "> =</ span > < span class ="n "> StrictMinMaxConstraint</ span > < span class ="p "> (</ span >
1085
- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> constraint_range</ span > < span class ="o "> .</ span > < span class ="n "> vr</ span > < span class ="o "> &</ span > < span class ="n "> ValueRanges</ span > < span class ="p "> (</ span > < span class ="n "> lower</ span > < span class ="o "> =</ span > < span class ="n "> lower</ span > < span class ="p "> ,</ span > < span class ="n "> upper</ span > < span class ="o "> =</ span > < span class ="n "> upper</ span > < span class ="p "> )</ span >
1087
+ < span class ="n "> vr</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> constraint_range</ span > < span class ="o "> .</ span > < span class ="n "> vr</ span > < span class ="o "> &</ span > < span class ="n "> ValueRanges</ span > < span class ="p "> (</ span > < span class ="n "> lower</ span > < span class ="o "> =</ span > < span class ="n "> lower</ span > < span class ="p "> ,</ span > < span class ="n "> upper</ span > < span class ="o "> =</ span > < span class ="n "> upper</ span > < span class ="p "> ),</ span >
1088
+ < span class ="n "> warn_only</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span >
1086
1089
< span class ="p "> )</ span >
1087
1090
< span class ="k "> return</ span > < span class ="n "> Constraint</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> w_tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> t_id</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> dim</ span > < span class ="p "> ,</ span > < span class ="n "> constraint_range</ span > < span class ="p "> )</ span >
1088
1091
0 commit comments