Nabla-Examples-and-Tests/73_ImageUploadBenchmark/app_resources/tile_upload.comp.hlsl at 717f6aef48ba948eb466db59c6236b73f1df9e78 · Devsh-Graphics-Programming/Nabla-Examples-and-Tests · GitHub

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
#include "common.hlsl"

[[vk::binding(0,0)]] RWTexture2D<float32_t4> dstImage;
[[vk::push_constant]] PushConstantData pc;

using namespace nbl::hlsl;

static const uint32_t TILE_SIZE = 128u;

[numthreads(128, 1, 1)]
[shader("compute")]
void SnakeStore(uint32_t3 ID : SV_DispatchThreadID)
{
    const uint32_t gIdx = ID.x;
    const uint32_t tileIdx = gIdx / (TILE_SIZE * TILE_SIZE);
    const uint32_t localIdx = gIdx % (TILE_SIZE * TILE_SIZE);
    const uint32_t2 tileBase = uint32_t2(tileIdx % pc.tilesPerRow, tileIdx / pc.tilesPerRow) * TILE_SIZE;
    const uint32_t2 localPos = uint32_t2(localIdx % TILE_SIZE, localIdx / TILE_SIZE);
    const uint32_t2 pixelPos = tileBase + localPos;

    const uint32_t packed = vk::RawBufferLoad<uint32_t>(pc.deviceBufferAddress + gIdx * 4u);
    dstImage[pixelPos] = unpackUnorm4x8(int32_t(packed));
}

[numthreads(128, 1, 1)]
[shader("compute")]
void SnakeLoad(uint32_t3 ID : SV_DispatchThreadID)
{
    const uint32_t gIdx = ID.x;
    const uint32_t tileIdx = gIdx / (TILE_SIZE * TILE_SIZE);
    const uint32_t localIdx = gIdx % (TILE_SIZE * TILE_SIZE);
    const uint32_t2 tileBase = uint32_t2(tileIdx % pc.tilesPerRow, tileIdx / pc.tilesPerRow) * TILE_SIZE;
    const uint32_t2 localPos = uint32_t2(localIdx % TILE_SIZE, localIdx / TILE_SIZE);
    const uint32_t2 pixelPos = tileBase + localPos;

    vk::RawBufferStore<uint32_t>(pc.deviceBufferAddress + gIdx * 4u, uint32_t(packUnorm4x8(dstImage[pixelPos])));
}

[numthreads(128, 1, 1)]
[shader("compute")]
void MortonStore(uint32_t3 ID : SV_DispatchThreadID)
{
    const uint32_t gIdx = ID.x;
    const uint32_t tileIdx = gIdx / (TILE_SIZE * TILE_SIZE);
    const uint32_t localIdx = gIdx % (TILE_SIZE * TILE_SIZE);
    const uint32_t2 tileBase = uint32_t2(tileIdx % pc.tilesPerRow, tileIdx / pc.tilesPerRow) * TILE_SIZE;

    morton::code<false, 7, 2> mc;
    mc.value = uint16_t(localIdx);
    const uint32_t2 localPos = _static_cast<uint32_t2>(mc);
    const uint32_t2 pixelPos = tileBase + localPos;

    const uint32_t packed = vk::RawBufferLoad<uint32_t>(pc.deviceBufferAddress + gIdx * 4u);
    dstImage[pixelPos] = unpackUnorm4x8(int32_t(packed));
}

[numthreads(128, 1, 1)]
[shader("compute")]
void MortonLoad(uint32_t3 ID : SV_DispatchThreadID)
{
    const uint32_t gIdx = ID.x;
    const uint32_t tileIdx = gIdx / (TILE_SIZE * TILE_SIZE);
    const uint32_t localIdx = gIdx % (TILE_SIZE * TILE_SIZE);
    const uint32_t2 tileBase = uint32_t2(tileIdx % pc.tilesPerRow, tileIdx / pc.tilesPerRow) * TILE_SIZE;

    morton::code<false, 7, 2> mc;
    mc.value = uint16_t(localIdx);
    const uint32_t2 localPos = _static_cast<uint32_t2>(mc);
    const uint32_t2 pixelPos = tileBase + localPos;

    vk::RawBufferStore<uint32_t>(pc.deviceBufferAddress + gIdx * 4u, uint32_t(packUnorm4x8(dstImage[pixelPos])));
}