1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// cuda-command-queue.h
#pragma once
#include "cuda-base.h"
#include "cuda-device.h"
#include "cuda-pipeline-state.h"
#include "cuda-shader-object.h"
#include "cuda-helper-functions.h"
namespace gfx
{
#ifdef GFX_ENABLE_CUDA
using namespace Slang;
namespace cuda
{
class CommandQueueImpl
: public ICommandQueue
, public ComObject
{
public:
SLANG_COM_OBJECT_IUNKNOWN_ALL
ICommandQueue* getInterface(const Guid& guid);
RefPtr<ComputePipelineStateImpl> currentPipeline;
RefPtr<RootShaderObjectImpl> currentRootObject;
RefPtr<DeviceImpl> renderer;
CUstream stream;
Desc m_desc;
void init(DeviceImpl* inRenderer);
~CommandQueueImpl();
virtual SLANG_NO_THROW const Desc& SLANG_MCALL getDesc() override { return m_desc; }
virtual SLANG_NO_THROW void SLANG_MCALL executeCommandBuffers(
GfxCount count, ICommandBuffer* const* commandBuffers, IFence* fence, uint64_t valueToSignal) override;
virtual SLANG_NO_THROW void SLANG_MCALL waitOnHost() override;
virtual SLANG_NO_THROW Result SLANG_MCALL waitForFenceValuesOnDevice(
GfxCount fenceCount, IFence** fences, uint64_t* waitValues) override;
virtual SLANG_NO_THROW Result SLANG_MCALL getNativeHandle(InteropHandle* outHandle) override;
void setPipelineState(IPipelineState* state);
Result bindRootShaderObject(IShaderObject* object);
void dispatchCompute(int x, int y, int z);
void copyBuffer(
IBufferResource* dst,
size_t dstOffset,
IBufferResource* src,
size_t srcOffset,
size_t size);
void uploadBufferData(IBufferResource* dst, size_t offset, size_t size, void* data);
void writeTimestamp(IQueryPool* pool, SlangInt index);
void execute(CommandBufferImpl* commandBuffer);
};
} // namespace cuda
#endif
} // namespace gfx
|
