yum-archive/TaSTT-Whisper

High-performance GPGPU inference of OpenAI's Whisper automatic speech recognition (ASR) model

git clone https://git.yummers.dev/yum-archive/TaSTT-Whisper

KonstantinSource codes8c4603c

master
branchesmasterall refs →
453 B22 linesraw
1#pragma once
2#include "../D3D/device.h"
3
4namespace DirectCompute
5{
6	class LookupTables
7	{
8		CComPtr<ID3D11ShaderResourceView> m_gelu, m_exponent;
9
10	public:
11
12		HRESULT create();
13		void clear();
14		ID3D11ShaderResourceView* gelu() const { return m_gelu; }
15		ID3D11ShaderResourceView* exponent() const { return m_exponent; }
16
17		__m128i getMemoryUsage() const;
18	};
19
20	// Singleton instance, defined in mlStartup.cpp
21	extern const LookupTables& lookupTables;
22}