Skip to content

Commit a1a553f

Browse files
committed
Add benchmarks
1 parent 47d1efd commit a1a553f

File tree

1 file changed

+43
-1
lines changed
  • lib/node_modules/@stdlib/ndarray/benchmark/python/numpy

1 file changed

+43
-1
lines changed

lib/node_modules/@stdlib/ndarray/benchmark/python/numpy/benchmark.py

+43-1
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def main():
206206
benchmark(name, setup, stmt, iterations)
207207

208208
name = "::3d,instantiation,dtype_cast"
209-
setup = "import numpy as np; x = np.array([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype='float32');"
209+
setup = "import numpy as np; x = np.array([[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]], dtype='float32');"
210210
stmt = "y = np.array(x, dtype='float64')"
211211
iterations = 100000
212212
benchmark(name, setup, stmt, iterations)
@@ -229,6 +229,48 @@ def main():
229229
iterations = 100000
230230
benchmark(name, setup, stmt, iterations)
231231

232+
name = "::4d,instantiation,ndarray"
233+
setup = "import numpy as np; x = np.array([[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]]);"
234+
stmt = "y = np.array(x)"
235+
iterations = 100000
236+
benchmark(name, setup, stmt, iterations)
237+
238+
name = "::4d,instantiation,no_cast"
239+
setup = "import numpy as np; x = np.array([[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]], dtype='float64');"
240+
stmt = "y = np.array(x, dtype='float64')"
241+
iterations = 100000
242+
benchmark(name, setup, stmt, iterations)
243+
244+
name = "::4d,instantiation,default_cast"
245+
setup = "import numpy as np; x = [[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]];"
246+
stmt = "y = np.array(x, dtype='float64')"
247+
iterations = 100000
248+
benchmark(name, setup, stmt, iterations)
249+
250+
name = "::4d,instantiation,dtype_cast"
251+
setup = "import numpy as np; x = np.array([[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]], dtype='float32');"
252+
stmt = "y = np.array(x, dtype='float64')"
253+
iterations = 100000
254+
benchmark(name, setup, stmt, iterations)
255+
256+
name = "::4d,instantiation:copy=false"
257+
setup = "import numpy as np; x = np.array([[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]]);"
258+
stmt = "y = np.array(x, copy=False)"
259+
iterations = 100000
260+
benchmark(name, setup, stmt, iterations)
261+
262+
name = "::4d,instantiation:copy=true"
263+
setup = "import numpy as np; x = np.array([[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]]);"
264+
stmt = "y = np.array(x, copy=True)"
265+
iterations = 100000
266+
benchmark(name, setup, stmt, iterations)
267+
268+
name = "::4d,instantiation:flatten=true"
269+
setup = "import numpy as np; x = [[[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]]];"
270+
stmt = "y = np.array(x)"
271+
iterations = 100000
272+
benchmark(name, setup, stmt, iterations)
273+
232274
print_summary(COUNT[0], COUNT[0])
233275

234276

0 commit comments

Comments
 (0)