Skip to content

Commit 4ae56f2

Browse files
committed
update allow_threads -> deatch
1 parent 29adc82 commit 4ae56f2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/lib.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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) {

0 commit comments

Comments
 (0)