File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1027,7 +1027,7 @@ fn optimize_greedy(
10271027 simplify : Option < bool > ,
10281028 use_ssa : Option < bool > ,
10291029) -> Vec < Vec < Node > > {
1030- py. allow_threads ( || {
1030+ py. detach ( || {
10311031 let n = inputs. len ( ) ;
10321032 let mut cp = ContractionProcessor :: new ( inputs, output, size_dict, false ) ;
10331033 if simplify. unwrap_or ( true ) {
@@ -1117,7 +1117,7 @@ fn optimize_random_greedy_track_flops(
11171117 simplify : Option < bool > ,
11181118 use_ssa : Option < bool > ,
11191119) -> ( Vec < Vec < Node > > , Score ) {
1120- py. allow_threads ( || {
1120+ py. detach ( || {
11211121 let ( costmod_min, costmod_max) = costmod. unwrap_or ( ( 0.1 , 4.0 ) ) ;
11221122 let costmod_diff = ( costmod_max - costmod_min) . abs ( ) ;
11231123 let is_const_costmod = costmod_diff < Score :: EPSILON ;
@@ -1262,7 +1262,7 @@ fn optimize_optimal(
12621262 simplify : Option < bool > ,
12631263 use_ssa : Option < bool > ,
12641264) -> Vec < Vec < Node > > {
1265- py. allow_threads ( || {
1265+ py. detach ( || {
12661266 let n = inputs. len ( ) ;
12671267 let mut cp = ContractionProcessor :: new ( inputs, output, size_dict, false ) ;
12681268 if simplify. unwrap_or ( true ) {
You can’t perform that action at this time.
0 commit comments