mlx/benchmarks/python
2025-04-17 13:53:11 -07:00
..
blas
comparative
batch_matmul_bench.py
compile_bench.py
conv1d_bench.py
conv2d_bench_cpu.py
conv2d_train_bench_cpu.py
conv2d_transpose_bench_cpu.py
conv3d_bench_cpu.py
conv3d_train_bench_cpu.py
conv3d_transpose_bench_cpu.py
conv_bench.py
conv_transpose_bench.py
distributed_bench.py
einsum_bench.py
fft_bench.py
gather_bench.py Remove unused modules (#1949) 2025-03-10 06:05:26 -07:00
gather_mm_bench.py Gather qmm batched kernel and refactoring of quantized (#2078) 2025-04-17 13:53:11 -07:00
gather_qmm_bench.py Gather qmm batched kernel and refactoring of quantized (#2078) 2025-04-17 13:53:11 -07:00
hadamard_bench.py
layer_norm_bench.py RMS norm without scaling (#1915) 2025-02-28 20:26:57 -08:00
rms_norm_bench.py RMS norm without scaling (#1915) 2025-02-28 20:26:57 -08:00
rope_bench.py
scatter_bench.py
sdpa_bench.py Support fused masking in Attention (#1924) 2025-03-20 11:01:32 -07:00
sdpa_vector_bench.py Allow different value dimensions in sdpa_vector (#1811) 2025-01-31 20:58:59 -08:00
single_ops.py
synchronize_bench.py Faster synchronization Fence primitive (#1773) 2025-01-17 18:42:19 -08:00
time_utils.py