@@ -10,7 +10,7 @@ use pyo3::{types::PyAnyMethods, Bound, IntoPyObjectExt, Python};
1010
1111#[ bench]
1212fn extract_success ( bencher : & mut Bencher ) {
13- Python :: with_gil ( |py| {
13+ Python :: attach ( |py| {
1414 let any = PyArray2 :: < f64 > :: zeros ( py, ( 10 , 10 ) , false ) . into_any ( ) ;
1515
1616 bencher. iter ( || {
@@ -23,7 +23,7 @@ fn extract_success(bencher: &mut Bencher) {
2323
2424#[ bench]
2525fn extract_failure ( bencher : & mut Bencher ) {
26- Python :: with_gil ( |py| {
26+ Python :: attach ( |py| {
2727 let any = PyArray2 :: < f64 > :: zeros ( py, ( 10 , 10 ) , false ) . into_any ( ) ;
2828
2929 bencher. iter ( || {
@@ -35,20 +35,20 @@ fn extract_failure(bencher: &mut Bencher) {
3535}
3636
3737#[ bench]
38- fn downcast_success ( bencher : & mut Bencher ) {
39- Python :: with_gil ( |py| {
38+ fn cast_success ( bencher : & mut Bencher ) {
39+ Python :: attach ( |py| {
4040 let any = PyArray2 :: < f64 > :: zeros ( py, ( 10 , 10 ) , false ) . into_any ( ) ;
4141
42- bencher. iter ( || black_box ( & any) . downcast :: < PyArray2 < f64 > > ( ) . unwrap ( ) ) ;
42+ bencher. iter ( || black_box ( & any) . cast :: < PyArray2 < f64 > > ( ) . unwrap ( ) ) ;
4343 } ) ;
4444}
4545
4646#[ bench]
47- fn downcast_failure ( bencher : & mut Bencher ) {
48- Python :: with_gil ( |py| {
47+ fn cast_failure ( bencher : & mut Bencher ) {
48+ Python :: attach ( |py| {
4949 let any = PyArray2 :: < f64 > :: zeros ( py, ( 10 , 10 ) , false ) . into_any ( ) ;
5050
51- bencher. iter ( || black_box ( & any) . downcast :: < PyArray2 < f64 > > ( ) . unwrap_err ( ) ) ;
51+ bencher. iter ( || black_box ( & any) . cast :: < PyArray2 < f64 > > ( ) . unwrap_err ( ) ) ;
5252 } ) ;
5353}
5454
@@ -63,7 +63,7 @@ impl Iterator for Iter {
6363}
6464
6565fn from_iter ( bencher : & mut Bencher , size : usize ) {
66- Python :: with_gil ( |py| {
66+ Python :: attach ( |py| {
6767 bencher. iter ( || {
6868 let iter = black_box ( Iter ( 0 ..size) ) ;
6969
@@ -90,7 +90,7 @@ fn from_iter_large(bencher: &mut Bencher) {
9090fn from_slice ( bencher : & mut Bencher , size : usize ) {
9191 let vec = ( 0 ..size) . collect :: < Vec < _ > > ( ) ;
9292
93- Python :: with_gil ( |py| {
93+ Python :: attach ( |py| {
9494 bencher. iter ( || {
9595 let slice = black_box ( & vec) ;
9696
@@ -115,13 +115,13 @@ fn from_slice_large(bencher: &mut Bencher) {
115115}
116116
117117fn from_object_slice ( bencher : & mut Bencher , size : usize ) {
118- let vec = Python :: with_gil ( |py| {
118+ let vec = Python :: attach ( |py| {
119119 ( 0 ..size)
120120 . map ( |val| val. into_py_any ( py) . unwrap ( ) )
121121 . collect :: < Vec < _ > > ( )
122122 } ) ;
123123
124- Python :: with_gil ( |py| {
124+ Python :: attach ( |py| {
125125 bencher. iter ( || {
126126 let slice = black_box ( & vec) ;
127127
@@ -148,7 +148,7 @@ fn from_object_slice_large(bencher: &mut Bencher) {
148148fn from_vec2 ( bencher : & mut Bencher , size : usize ) {
149149 let vec2 = vec ! [ vec![ 0 ; size] ; size] ;
150150
151- Python :: with_gil ( |py| {
151+ Python :: attach ( |py| {
152152 bencher. iter ( || {
153153 let vec2 = black_box ( & vec2) ;
154154
@@ -175,7 +175,7 @@ fn from_vec2_large(bencher: &mut Bencher) {
175175fn from_vec3 ( bencher : & mut Bencher , size : usize ) {
176176 let vec3 = vec ! [ vec![ vec![ 0 ; size] ; size] ; size] ;
177177
178- Python :: with_gil ( |py| {
178+ Python :: attach ( |py| {
179179 bencher. iter ( || {
180180 let vec3 = black_box ( & vec3) ;
181181
0 commit comments