yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

Harsh Aggarwal (NVIDIA)Fix 7723 - Add autodiff tests (#7919)d10732742

master
branchesmasterall refs →
1008 B34 linesraw
1//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type
2//TEST(compute):COMPARE_COMPUTE_EX:-cuda -compute -shaderobj -output-using-type
3
4//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer
5RWStructuredBuffer<float> outputBuffer;
6
7typedef DifferentialPair<float> dpfloat;
8typedef float.Differential dfloat;
9
10import test_intrinsics;
11
12dpfloat my_pow_jvp(dpfloat x, dpfloat n)
13{
14    return dpfloat(
15        pow(x.p, n.p),
16        x.d * n.p * pow(x.p, n.p-1) + n.d * pow(x.p, n.p) * log(x.p));
17}
18
19[ForwardDerivative(my_pow_jvp)]
20float _pow(float, float);
21
22[numthreads(1, 1, 1)]
23void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
24{
25     {
26        dpfloat dpa = dpfloat(5.0, 1.0);
27        dpfloat dpn = dpfloat(2, 0.0);
28
29        outputBuffer[0] = __fwd_diff(_pow)(dpa, dpn).d;        // Expect: 10.0
30        outputBuffer[1] = __fwd_diff(_pow)(
31            dpfloat(dpa.p, 0.0),
32            dpfloat(dpn.p, 1.0)).d;                     // Expect: 40.23595
33    }
34}