|
31 | 31 | import dpnp
|
32 | 32 | import numpy as np
|
33 | 33 |
|
34 |
| -ROUNDS=30 |
35 |
| -ITERATIONS=4 |
| 34 | +ROUNDS = 30 |
| 35 | +ITERATIONS = 4 |
| 36 | + |
| 37 | +NNUMBERS = 2**26 |
36 | 38 |
|
37 |
| -NNUMBERS=2**26 |
38 | 39 |
|
39 | 40 | @pytest.mark.parametrize("function", [dpnp.random.beta, np.random.beta],
|
40 |
| - ids=["dpnp", "numpy"]) |
| 41 | + ids=["dpnp", "numpy"]) |
41 | 42 | def test_beta(benchmark, function):
|
42 | 43 | result = benchmark.pedantic(target=function, args=(4.0, 5.0, NNUMBERS,),
|
43 | 44 | rounds=ROUNDS, iterations=ITERATIONS)
|
44 | 45 |
|
45 | 46 |
|
46 | 47 | @pytest.mark.parametrize("function", [dpnp.random.exponential, np.random.exponential],
|
47 |
| - ids=["dpnp", "numpy"]) |
| 48 | + ids=["dpnp", "numpy"]) |
48 | 49 | def test_exponential(benchmark, function):
|
49 | 50 | result = benchmark.pedantic(target=function, args=(4.0, NNUMBERS,),
|
50 | 51 | rounds=ROUNDS, iterations=ITERATIONS)
|
51 | 52 |
|
52 | 53 |
|
53 | 54 | @pytest.mark.parametrize("function", [dpnp.random.gamma, np.random.gamma],
|
54 |
| - ids=["dpnp", "numpy"]) |
| 55 | + ids=["dpnp", "numpy"]) |
55 | 56 | def test_gamma(benchmark, function):
|
56 | 57 | result = benchmark.pedantic(target=function, args=(2.0, 4.0, NNUMBERS,),
|
57 | 58 | rounds=ROUNDS, iterations=ITERATIONS)
|
58 | 59 |
|
59 | 60 |
|
60 | 61 | @pytest.mark.parametrize("function", [dpnp.random.normal, np.random.normal],
|
61 |
| - ids=["dpnp", "numpy"]) |
| 62 | + ids=["dpnp", "numpy"]) |
62 | 63 | def test_normal(benchmark, function):
|
63 | 64 | result = benchmark.pedantic(target=function, args=(0.0, 1.0, NNUMBERS,),
|
64 | 65 | rounds=ROUNDS, iterations=ITERATIONS)
|
65 | 66 |
|
66 | 67 |
|
67 | 68 | @pytest.mark.parametrize("function", [dpnp.random.uniform, np.random.uniform],
|
68 |
| - ids=["dpnp", "numpy"]) |
| 69 | + ids=["dpnp", "numpy"]) |
69 | 70 | def test_uniform(benchmark, function):
|
70 | 71 | result = benchmark.pedantic(target=function, args=(0.0, 1.0, NNUMBERS,),
|
71 | 72 | rounds=ROUNDS, iterations=ITERATIONS)
|
0 commit comments