diff options
| author | Yong He <yonghe@outlook.com> | 2022-10-27 12:19:30 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-27 12:19:30 -0700 |
| commit | 0cbef6fd6d7924d37ef3ea5ec7c848c80947d13f (patch) | |
| tree | 173fa18c39638e7d41ae092b9012554cb867a31b /tests/autodiff/vector-arithmetic-jvp.slang | |
| parent | 351e78f3abc54f114237d4af64f8199476ebf176 (diff) | |
Rename `__jvp`-->`__fwd_diff`. (#2471)
Co-authored-by: Yong He <yhe@nvidia.com>
Diffstat (limited to 'tests/autodiff/vector-arithmetic-jvp.slang')
| -rw-r--r-- | tests/autodiff/vector-arithmetic-jvp.slang | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/autodiff/vector-arithmetic-jvp.slang b/tests/autodiff/vector-arithmetic-jvp.slang index e05d94733..b79b3e764 100644 --- a/tests/autodiff/vector-arithmetic-jvp.slang +++ b/tests/autodiff/vector-arithmetic-jvp.slang @@ -48,17 +48,17 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) float4 a4 = float4(2.0, 1.0, 0.0, 2.0); float4 b4 = float4(1.5, -2.0, 1.0, 1.5); - outputBuffer[0] = __jvp(f)(dpfloat3(a, da)).d().z; // Expect: 1 + outputBuffer[0] = __fwd_diff(f)(dpfloat3(a, da)).d().z; // Expect: 1 - outputBuffer[1] = __jvp(g)( + outputBuffer[1] = __fwd_diff(g)( dpfloat3(a, da), dpfloat3(b, float3(2.0, 1.0, 0.0))).d().y; // Expect: 8 - outputBuffer[2] = __jvp(h)( + outputBuffer[2] = __fwd_diff(h)( dpfloat2(a2, float2(1.0, 0.0)), dpfloat2(b2, float2(1.0, 1.0))).d().x; // Expect: 8 - outputBuffer[3] = __jvp(j)( + outputBuffer[3] = __fwd_diff(j)( dpfloat4(a4, float4(1.0)), dpfloat4(b4, float4(2.0))).d().w; // Expect: 9 } |