ThreadAbort/IndexTTS-Rust

1//! Benchmark for mel-spectrogram computation2 3use criterion::{black_box, criterion_group, criterion_main, Criterion};4use indextts::audio::{mel_spectrogram, AudioConfig};5 6fn bench_mel_spectrogram(c: &mut Criterion) {7    let config = AudioConfig::default();8 9    // Generate 1 second of audio10    let num_samples = config.sample_rate as usize;11    let signal: Vec<f32> = (0..num_samples).map(|i| (i as f32 * 0.01).sin()).collect();12 13    c.bench_function("mel_spectrogram_1s", |b| {14        b.iter(|| mel_spectrogram(black_box(&signal), black_box(&config)))15    });16 17    // Generate 10 seconds of audio18    let long_signal: Vec<f32> = (0..num_samples * 10)19        .map(|i| (i as f32 * 0.01).sin())20        .collect();21 22    c.bench_function("mel_spectrogram_10s", |b| {23        b.iter(|| mel_spectrogram(black_box(&long_signal), black_box(&config)))24    });25}26 27fn bench_stft(c: &mut Criterion) {28    let config = AudioConfig::default();29    let num_samples = config.sample_rate as usize;30    let signal: Vec<f32> = (0..num_samples).map(|i| (i as f32 * 0.01).sin()).collect();31 32    c.bench_function("stft_1s", |b| {33        b.iter(|| {34            indextts::audio::mel::stft(35                black_box(&signal),36                black_box(config.n_fft),37                black_box(config.hop_length),38                black_box(config.win_length),39            )40        })41    });42}43 44criterion_group!(benches, bench_mel_spectrogram, bench_stft);45criterion_main!(benches);46