blob: 1434f1a66b7b1655257c67faadbcbc342bcbaddb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
// A compute shader to add gradients to a mip-map texture.
cbuffer Uniforms
{
uint dstWidth;
uint dstHeight;
float learningRate;
RWTexture2D dstTexture;
RWTexture2D srcTexture;
}
[shader("compute")]
[numthreads(16, 16, 1)]
void computeMain(uint3 threadIdx : SV_DispatchThreadID)
{
uint x = threadIdx.x;
uint y = threadIdx.y;
if (x >= dstWidth) return;
if (y >= dstHeight) return;
var val = srcTexture[uint2(x, y)];
dstTexture[uint2(x, y)] = float4((dstTexture[uint2(x, y)] - val * learningRate).xyz, 1.0);
}
|
