Angelos Katharopoulos 5de6d94a90 Gather qmm batched kernel and refactoring of quantized (#2078)

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

…

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_bench.py

…

rope_bench.py

…

scatter_bench.py

…

sdpa_bench.py

…

sdpa_vector_bench.py

…

single_ops.py

…

synchronize_bench.py

…

time_utils.py

…