| 43 | } |
| 44 | |
| 45 | void time_random_generation() { |
| 46 | int M = 2000; |
| 47 | int N = 500; |
| 48 | |
| 49 | auto uniform = [&]() { return mx::random::uniform({M, N}, mx::float32); }; |
| 50 | TIME(uniform); |
| 51 | auto normal = [&]() { return mx::random::normal({M, N}, mx::float32); }; |
| 52 | TIME(normal); |
| 53 | } |
| 54 | |
| 55 | void time_unary_ops() { |
| 56 | int M = 2000; |