Whisper/ML/LookupTables.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#pragma once
#include "../D3D/device.h"

namespace DirectCompute
{
	class LookupTables
	{
		CComPtr<ID3D11ShaderResourceView> m_gelu, m_exponent;

	public:

		HRESULT create();
		void clear();
		ID3D11ShaderResourceView* gelu() const { return m_gelu; }
		ID3D11ShaderResourceView* exponent() const { return m_exponent; }

		__m128i getMemoryUsage() const;
	};

	// Singleton instance, defined in mlStartup.cpp
	extern const LookupTables& lookupTables;
}