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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
//TEST:EXECUTABLE:
import gfx;
import slang;
export __extern_cpp int main()
{
gfx.DeviceDesc deviceDesc = {};
deviceDesc.deviceType = gfx.DeviceType.CPU;
Optional<gfx.IDevice> device;
gfx.gfxCreateDevice(&deviceDesc, device);
if (device == none)
{
printf("fail\n");
return -1;
}
gfx.CommandQueueDesc queueDesc = {gfx::QueueType::Graphics};
queueDesc.type = gfx.QueueType.Graphics;
Optional<gfx.ICommandQueue> queue;
device.value.createCommandQueue(&queueDesc, queue);
gfx.ShaderProgramDesc2 programDesc = {};
NativeString s = R"(
[shader("compute")]
[numthreads(4, 1, 1)]
void computeMain(
uint3 sv_dispatchThreadID: SV_DispatchThreadID,
uniform RWStructuredBuffer<float> buffer
)
{
var input = buffer[sv_dispatchThreadID.x];
buffer[sv_dispatchThreadID.x] = sv_dispatchThreadID.x;
})";
programDesc.sourceData = s;
programDesc.sourceType = gfx.ShaderModuleSourceType.SlangSource;
programDesc.sourceDataSize = s.length;
programDesc.entryPointCount = 1;
NativeString entryPointName = "computeMain";
programDesc.entryPointNames = &entryPointName;
Optional<gfx.IShaderProgram> program;
Optional<slang.ISlangBlob> diagBlob;
device.value.createProgram2(&programDesc, program, diagBlob);
Optional<gfx.IPipelineState> pipeline;
gfx.ComputePipelineStateDesc pipelineDesc;
pipelineDesc.program = NativeRef<gfx.IShaderProgram>(program.value);
device.value.createComputePipelineState(&pipelineDesc, pipeline);
Optional<gfx.ITransientResourceHeap> transientHeap;
gfx.TransientResourceHeapDesc transientHeapDesc;
transientHeapDesc.constantBufferDescriptorCount = 64;
transientHeapDesc.constantBufferSize = 1024;
transientHeapDesc.srvDescriptorCount = 1024;
transientHeapDesc.uavDescriptorCount = 1024;
transientHeapDesc.samplerDescriptorCount = 256;
transientHeapDesc.accelerationStructureDescriptorCount = 32;
device.value.createTransientResourceHeap(&transientHeapDesc, transientHeap);
Optional<gfx.IBufferResource> buffer;
gfx.BufferResourceDesc bufferDesc = {};
bufferDesc.memoryType = gfx.MemoryType.DeviceLocal;
bufferDesc.allowedStates.add(gfx.ResourceState.UnorderedAccess);
bufferDesc.defaultState = gfx.ResourceState.UnorderedAccess;
bufferDesc.elementSize = 4;
bufferDesc.sizeInBytes = 256;
bufferDesc.type = gfx.ResourceType.Buffer;
device.value.createBufferResource(&bufferDesc, nullptr, buffer);
Optional<gfx.IResourceView> bufferView;
gfx.ResourceViewDesc viewDesc;
viewDesc.type = gfx.ResourceViewType.UnorderedAccess;
device.value.createBufferView(buffer.value, none, &viewDesc, bufferView);
Optional<gfx.ICommandBuffer> commandBuffer;
transientHeap.value.createCommandBuffer(commandBuffer);
Optional<gfx.IComputeCommandEncoder> encoder;
commandBuffer.value.encodeComputeCommands(encoder);
Optional<gfx.IShaderObject> rootObject;
encoder.value.bindPipeline(pipeline.value, rootObject);
Optional<gfx.IShaderObject> entryPointObject;
rootObject.value.getEntryPoint(0, entryPointObject);
gfx.ShaderOffset offset = {};
entryPointObject.value.setResource(&offset, bufferView.value);
encoder.value.dispatchCompute(1, 1, 1);
encoder.value.endEncoding();
commandBuffer.value.close();
NativeRef<gfx.ICommandBuffer> commandBufferRef = NativeRef<gfx.ICommandBuffer>(commandBuffer.value);
queue.value.executeCommandBuffers(1, &commandBufferRef, none, 0);
queue.value.waitOnHost();
Optional<slang.ISlangBlob> blob;
device.value.readBufferResource(buffer.value, 0, 16, blob);
for (int i = 0; i < 4; i++)
{
float val = ((float *)blob.value.getBufferPointer())[i];
printf("%.1f\n", val);
}
return 0;
}
|
