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

KonstantinPerformance improvement, `softMax` shader27dfc34

master
1.0 KiB55 linesraw
1// This source file is generated by a tool
2#include "stdafx.h"
3#include "shaderNames.h"
4
5static const std::array<const char*, 40> s_shaderNames = 
6{
7	"add",
8	"addInPlace",
9	"addRepeat",
10	"addRepeatEx",
11	"addRepeatGelu",
12	"addRepeatScale",
13	"addRows",
14	"convolutionMain",
15	"convolutionMain2",
16	"convolutionMain2Fixed",
17	"convolutionPrep1",
18	"convolutionPrep2",
19	"copyConvert",
20	"copyTranspose",
21	"diagMaskInf",
22	"flashAttention",
23	"flashAttentionCompat1",
24	"flashAttentionCompat2",
25	"flashAttentionCompat3",
26	"fmaRepeat1",
27	"fmaRepeat2",
28	"matReshapePanels",
29	"mulMatByRow",
30	"mulMatByRowTiled",
31	"mulMatByRowTiledEx",
32	"mulMatByScalar",
33	"mulMatDotMain",
34	"mulMatDotReshape",
35	"mulMatMadMain",
36	"mulMatTiled",
37	"mulMatTiledEx",
38	"norm",
39	"normCompat",
40	"normFixed",
41	"scaleInPlace",
42	"softMax",
43	"softMaxCompat",
44	"softMaxFixed",
45	"softMaxLong",
46	"zeroMemory",
47};
48
49const char* DirectCompute::computeShaderName( eComputeShader cs )
50{
51	const uint16_t i = (uint16_t)cs;
52	if( i < s_shaderNames.size() )
53		return s_shaderNames[ i ];
54	return nullptr;
55}