ffef323c4c
* whisper : use polymorphic class to calculate mel spectrogram * whisper : add cuda-specific mel spectrogram calculation * whisper : conditionally compile cufftGetErrorString to avoid warnings * build : add new files to makefile * ruby : add new files to conf script * build : fix typo in makefile * whisper : suppress cub warning for deprecated C++ std in whisper-mel-cuda
34 lines
706 B
C++
34 lines
706 B
C++
#pragma once
|
|
#include "ggml-backend.h"
|
|
#include <vector>
|
|
|
|
struct whisper_mel {
|
|
int n_len;
|
|
int n_len_org;
|
|
int n_mel;
|
|
|
|
std::vector<float> data;
|
|
};
|
|
|
|
struct whisper_filters {
|
|
int32_t n_mel;
|
|
int32_t n_fft;
|
|
|
|
std::vector<float> data;
|
|
};
|
|
|
|
template <typename T>
|
|
struct whisper_span {
|
|
T * data;
|
|
int len;
|
|
};
|
|
|
|
struct whisper_mel_calc {
|
|
virtual ~whisper_mel_calc();
|
|
virtual whisper_mel calculate(whisper_span<const float> samples, int n_threads) const = 0;
|
|
static whisper_span<const float> hann_window();
|
|
};
|
|
|
|
// returns a new pointer which needs to be freed with delete
|
|
whisper_mel_calc * whisper_mel_calc_create(ggml_backend_t backend, const whisper_filters & filters);
|