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 →
693 B23 linesraw
1#include "stdafx.h"
2#include "TensorGpuViews.h"
3using namespace DirectCompute;
4
5HRESULT TensorGpuViews::create( ID3D11Buffer* gpuBuffer, DXGI_FORMAT format, size_t countElements, bool makeUav )
6{
7	srv = nullptr;
8	uav = nullptr;
9
10	if( countElements > UINT_MAX )
11		return DISP_E_OVERFLOW;
12
13	CD3D11_SHADER_RESOURCE_VIEW_DESC viewDesc{ D3D11_SRV_DIMENSION_BUFFER, format, 0, (UINT)countElements };
14	CHECK( device()->CreateShaderResourceView( gpuBuffer, &viewDesc, &srv ) );
15
16	if( makeUav )
17	{
18		CD3D11_UNORDERED_ACCESS_VIEW_DESC uavDesc{ D3D11_UAV_DIMENSION_BUFFER, format , 0, (UINT)countElements };
19		CHECK( device()->CreateUnorderedAccessView( gpuBuffer, &uavDesc, &uav ) );
20	}
21
22	return S_OK;
23}