Rename `__jvp`-->`__fwd_diff`. (#2471)

Co-authored-by: Yong He <yhe@nvidia.com>

18 files changed, 45 insertions, 45 deletions

diff --git a/tests/autodiff/arithmetic-jvp.slang b/tests/autodiff/arithmetic-jvp.slang
index ddd1a4aa9..0c7dd039d 100644
--- a/tests/autodiff/arithmetic-jvp.slang
+++ b/tests/autodiff/arithmetic-jvp.slang
@@ -43,10 +43,10 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat dpa = dpfloat(2.0, 1.0);
         dpfloat dpb = dpfloat(1.5, 1.0);
 
-        outputBuffer[0] = __jvp(f)(dpa).d();                    // Expect: 1
-        outputBuffer[1] = __jvp(f)(dpfloat(dpa.p(), 0.0)).d();  // Expect: 0
-        outputBuffer[2] = __jvp(g)(dpa).d();                    // Expect: 2
-        outputBuffer[3] = __jvp(h)(dpa, dpb).d();               // Expect: 8
-        outputBuffer[4] = __jvp(j)(dpa, dpb).d();               // Expect: 1
+        outputBuffer[0] = __fwd_diff(f)(dpa).d();                   // Expect: 1
+        outputBuffer[1] = __fwd_diff(f)(dpfloat(dpa.p(), 0.0)).d(); // Expect: 0
+        outputBuffer[2] = __fwd_diff(g)(dpa).d();                   // Expect: 2
+        outputBuffer[3] = __fwd_diff(h)(dpa, dpb).d();              // Expect: 8
+        outputBuffer[4] = __fwd_diff(j)(dpa, dpb).d();               // Expect: 1
     }
 }
diff --git a/tests/autodiff/auto-differential-type.slang b/tests/autodiff/auto-differential-type.slang
index f15fb6417..b551db4ab 100644
--- a/tests/autodiff/auto-differential-type.slang
+++ b/tests/autodiff/auto-differential-type.slang
@@ -14,7 +14,7 @@ struct A : IDifferentiable
     [__unsafeForceInlineEarly]
     static Differential zero()
     {
-        Differential b = {0.0, 0.0};
+        Differential b = {0.0, float.zero()};
         return b;
     }
 
@@ -53,6 +53,6 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
 
         dpA dpa = dpA(a, b);
 
-        outputBuffer[0] = __jvp(f)(dpa).d().x;                    // Expect: 1
+        outputBuffer[0] = __fwd_diff(f)(dpa).d().x;                    // Expect: 1
     }
 }
diff --git a/tests/autodiff/custom-intrinsic.slang b/tests/autodiff/custom-intrinsic.slang
index 8ce354edc..02f6541f5 100644
--- a/tests/autodiff/custom-intrinsic.slang
+++ b/tests/autodiff/custom-intrinsic.slang
@@ -108,12 +108,12 @@ void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID)
         dpfloat dpa = dpfloat(2.0, 1.0);
 
         outputBuffer[0] = f(dpa.p());        // Expect: 7.389056
-        outputBuffer[1] = __jvp(f)(dpa).d(); // Expect: 7.389056
+        outputBuffer[1] = __fwd_diff(f)(dpa).d(); // Expect: 7.389056
 
         // g() needs additional handling of  IRMakeDifferentialPair(PtrType). This needs to 
         // generate a new var, load from the individual vars and store into the pair var.
 
         //outputBuffer[2] = g(dpa.p());        // Expect: 1.381773
-        //outputBuffer[3] = __jvp(g)(dpa).d(); // Expect: -0.301168
+        //outputBuffer[3] = __fwd_diff(g)(dpa).d(); // Expect: -0.301168
     }
 }
\ No newline at end of file
diff --git a/tests/autodiff/differential-method-synthesis.slang b/tests/autodiff/differential-method-synthesis.slang
index 53957fd91..3ecd636e9 100644
--- a/tests/autodiff/differential-method-synthesis.slang
+++ b/tests/autodiff/differential-method-synthesis.slang
@@ -40,7 +40,7 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         A a = {1.0, 2.0};
         A.Differential b = {0.2};
         dpA dpa = dpA(a, b);
-        outputBuffer[0] = __jvp(f)(dpa).d().b.x; // Expect: 0
+        outputBuffer[0] = __fwd_diff(f)(dpa).d().b.x; // Expect: 0
         outputBuffer[1] = A.dadd(b, b).b.x;      // Expect: 0.4
         outputBuffer[2] = A.dmul(a, b).b.x;      // Expect: 0.2
     }
diff --git a/tests/autodiff/dstdlib.slang b/tests/autodiff/dstdlib.slang
index 6c7ecffbe..614de54f6 100644
--- a/tests/autodiff/dstdlib.slang
+++ b/tests/autodiff/dstdlib.slang
@@ -28,10 +28,10 @@ void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID)
         dpfloat dpa = dpfloat(2.0, 1.0);
 
         outputBuffer[0] = f(dpa.p());        // Expect: 7.389056
-        outputBuffer[1] = __jvp(f)(dpa).d(); // Expect: 7.389056
+        outputBuffer[1] = __fwd_diff(f)(dpa).d(); // Expect: 7.389056
         outputBuffer[2] = g(dpa.p());        // Expect: 0.909297 
-        outputBuffer[3] = __jvp(g)(dpa).d(); // Expect: -0.416146
+        outputBuffer[3] = __fwd_diff(g)(dpa).d(); // Expect: -0.416146
         outputBuffer[4] = h(dpa.p());        // Expect: -0.416146
-        outputBuffer[5] = __jvp(h)(dpa).d(); // Expect: -0.909297 
+        outputBuffer[5] = __fwd_diff(h)(dpa).d(); // Expect: -0.909297 
     }
 }
\ No newline at end of file
diff --git a/tests/autodiff/generic-custom-jvp.slang b/tests/autodiff/generic-custom-jvp.slang
index 3f0d85b60..f0b8d3898 100644
--- a/tests/autodiff/generic-custom-jvp.slang
+++ b/tests/autodiff/generic-custom-jvp.slang
@@ -27,8 +27,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat dpa = dpfloat(5.0, 1.0);
         dpfloat dpn = dpfloat(2, 0.0);
 
-        outputBuffer[0] = __jvp(_pow)(dpa, dpn).d();        // Expect: 10.0
-        outputBuffer[1] = __jvp(_pow)(
+        outputBuffer[0] = __fwd_diff(_pow)(dpa, dpn).d();        // Expect: 10.0
+        outputBuffer[1] = __fwd_diff(_pow)(
             dpfloat(dpa.p(), 0.0),
             dpfloat(dpn.p(), 1.0)).d();                     // Expect: 40.23595
     }
diff --git a/tests/autodiff/generic-impl-jvp.slang b/tests/autodiff/generic-impl-jvp.slang
index fe4ffc426..e14f851ac 100644
--- a/tests/autodiff/generic-impl-jvp.slang
+++ b/tests/autodiff/generic-impl-jvp.slang
@@ -283,8 +283,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat3 dpf3 = dpfloat3(myfloat3(float3(1.0, 3.0, 5.0)), mydfloat3(float3(0.5, 1.5, 2.5)));
 
         outputBuffer[0] = f(dpa.p());                               // Expect: 22.0
-        outputBuffer[1] = __jvp(f)(dpfloat(2.0, 0.5)).d();          // Expect: 9.5
-        outputBuffer[2] = __jvp(f)(dpf4).d().val.values[3];         // Expect: 27.5
-        outputBuffer[3] = __jvp(f)(dpf3).d().val.values[1];         // Expect: 40.5
+        outputBuffer[1] = __fwd_diff(f)(dpfloat(2.0, 0.5)).d();          // Expect: 9.5
+        outputBuffer[2] = __fwd_diff(f)(dpf4).d().val.values[3];         // Expect: 27.5
+        outputBuffer[3] = __fwd_diff(f)(dpf3).d().val.values[1];         // Expect: 40.5
     }
 }
diff --git a/tests/autodiff/generic-jvp.slang b/tests/autodiff/generic-jvp.slang
index bcd5e764e..365be45aa 100644
--- a/tests/autodiff/generic-jvp.slang
+++ b/tests/autodiff/generic-jvp.slang
@@ -189,8 +189,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat3 dpf3 = dpfloat3(myfloat3(float3(1.0, 3.0, 5.0)), myfloat3(float3(0.5, 1.5, 2.5)));
 
         outputBuffer[0] = f(dpa.p());                       // Expect: 22.0
-        outputBuffer[1] = __jvp(f)(dpfloat(2.0, 0.5)).d();  // Expect: 9.5
-        outputBuffer[2] = __jvp(f)(dpf4).d().val.w;         // Expect: 27.5
-        outputBuffer[3] = __jvp(f)(dpf3).d().val.y;         // Expect: 40.5
+        outputBuffer[1] = __fwd_diff(f)(dpfloat(2.0, 0.5)).d();  // Expect: 9.5
+        outputBuffer[2] = __fwd_diff(f)(dpf4).d().val.w;         // Expect: 27.5
+        outputBuffer[3] = __fwd_diff(f)(dpf3).d().val.y;         // Expect: 40.5
     }
 }
diff --git a/tests/autodiff/getter-setter-multi.slang b/tests/autodiff/getter-setter-multi.slang
index c19a3f6bb..08816c5bc 100644
--- a/tests/autodiff/getter-setter-multi.slang
+++ b/tests/autodiff/getter-setter-multi.slang
@@ -66,8 +66,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
 
         dpA dpa = dpA(a, b);
 
-        outputBuffer[0] = __jvp(f)(dpa).d().z.z;  // Expect: 0.5
-        outputBuffer[1] = __jvp(f)(dpa).d().k[5]; // Expect: 1
-        outputBuffer[2] = __jvp(f)(dpa).d().k[2]; // Expect: 1.5
+        outputBuffer[0] = __fwd_diff(f)(dpa).d().z.z;  // Expect: 0.5
+        outputBuffer[1] = __fwd_diff(f)(dpa).d().k[5]; // Expect: 1
+        outputBuffer[2] = __fwd_diff(f)(dpa).d().k[2]; // Expect: 1.5
     }
 }
diff --git a/tests/autodiff/getter-setter.slang b/tests/autodiff/getter-setter.slang
index 0e8cac13b..2f385b87f 100644
--- a/tests/autodiff/getter-setter.slang
+++ b/tests/autodiff/getter-setter.slang
@@ -59,6 +59,6 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
 
         dpA dpa = dpA(a, b);
 
-        outputBuffer[0] = __jvp(f)(dpa).d().z;                    // Expect: 1
+        outputBuffer[0] = __fwd_diff(f)(dpa).d().z;                    // Expect: 1
     }
 }
diff --git a/tests/autodiff/imported-custom-jvp.slang b/tests/autodiff/imported-custom-jvp.slang
index ee8bdf51d..8adcdee25 100644
--- a/tests/autodiff/imported-custom-jvp.slang
+++ b/tests/autodiff/imported-custom-jvp.slang
@@ -20,6 +20,6 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat dpa = dpfloat(2.0, 1.0);
         dpfloat dpb = dpfloat(1.5, 1.0);
 
-        outputBuffer[0] = __jvp(f)(dpa).d();                      // Expect: 2
+        outputBuffer[0] = __fwd_diff(f)(dpa).d();                      // Expect: 2
     }
 }
diff --git a/tests/autodiff/inout-parameters-jvp.slang b/tests/autodiff/inout-parameters-jvp.slang
index ba04c6b65..e53e5db7c 100644
--- a/tests/autodiff/inout-parameters-jvp.slang
+++ b/tests/autodiff/inout-parameters-jvp.slang
@@ -33,12 +33,12 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
 
     dpfloat dpz = dpfloat(z, dz);
 
-    __jvp(h)(dpfloat(x, dx), dpfloat(y, dy), dpz);
+    __fwd_diff(h)(dpfloat(x, dx), dpfloat(y, dy), dpz);
 
     outputBuffer[0] = dpz.d(); // Expect: 12.0
     outputBuffer[1] = dpz.p(); // Expect: 6.75
 
-    __jvp(g)(dpfloat(x, dx), dpfloat(y, dy), dpz);
+    __fwd_diff(g)(dpfloat(x, dx), dpfloat(y, dy), dpz);
 
     outputBuffer[2] = dpz.d(); // Expect: 21.5
     outputBuffer[3] = dpz.p(); // Expect: 12.5
diff --git a/tests/autodiff/local-redecl-custom-jvp.slang b/tests/autodiff/local-redecl-custom-jvp.slang
index 6241a8bf5..79b90bd16 100644
--- a/tests/autodiff/local-redecl-custom-jvp.slang
+++ b/tests/autodiff/local-redecl-custom-jvp.slang
@@ -25,8 +25,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         dpfloat dpa = dpfloat(5.0, 1.0);
         dpfloat dpn = dpfloat(2, 0.0);
 
-        outputBuffer[0] = __jvp(_pow)(dpa, dpn).d();        // Expect: 10.0
-        outputBuffer[1] = __jvp(_pow)(
+        outputBuffer[0] = __fwd_diff(_pow)(dpa, dpn).d();        // Expect: 10.0
+        outputBuffer[1] = __fwd_diff(_pow)(
             dpfloat(dpa.p(), 0.0),
             dpfloat(dpn.p(), 1.0)).d();                     // Expect: 40.23595
     }
diff --git a/tests/autodiff/nested-jvp.slang b/tests/autodiff/nested-jvp.slang
index baebeee56..96648d861 100644
--- a/tests/autodiff/nested-jvp.slang
+++ b/tests/autodiff/nested-jvp.slang
@@ -58,7 +58,7 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         float3 d_f90 = float3(0.9, 0.9, 0.9);
         float d_cosTheta = 1.0;
 
-        outputBuffer[0] = __jvp(fresnel)(
+        outputBuffer[0] = __fwd_diff(fresnel)(
             dpfloat3(f0, d_f0),
             dpfloat3(f90, d_f90),
             dpfloat(cosTheta, d_cosTheta)).d().y; // Expect: -0.031250
@@ -71,14 +71,14 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
         float db = -1.0;
         float dc = 0.2;
 
-        outputBuffer[1] = __jvp(g)(
+        outputBuffer[1] = __fwd_diff(g)(
             dpfloat(a, da),
             dpfloat(b, db),
             dpfloat(c, dc)).d();                 // Expect: -0.24375
 
         outputBuffer[2] = g(a, b, c);       // Expect: 0.95625
 
-        outputBuffer[3] = __jvp(g)(
+        outputBuffer[3] = __fwd_diff(g)(
             dpfloat(a, da),
             dpfloat(b, db),
             dpfloat(3.0, dc)).d();               // Expect: -0.4;
diff --git a/tests/autodiff/out-parameters-jvp.slang b/tests/autodiff/out-parameters-jvp.slang
index b243d4fb5..9a311ed31 100644
--- a/tests/autodiff/out-parameters-jvp.slang
+++ b/tests/autodiff/out-parameters-jvp.slang
@@ -23,7 +23,7 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
     float dy = 0.5;
 
     dpfloat dresult;
-    __jvp(h)(dpfloat(x, dx), dpfloat(y, dy), dresult);
+    __fwd_diff(h)(dpfloat(x, dx), dpfloat(y, dy), dresult);
 
     outputBuffer[0] = dresult.d(); // Expect: 9.5
 
diff --git a/tests/autodiff/overloads-jvp.slang b/tests/autodiff/overloads-jvp.slang
index 26b5c0076..95b9cadd3 100644
--- a/tests/autodiff/overloads-jvp.slang
+++ b/tests/autodiff/overloads-jvp.slang
@@ -33,8 +33,8 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
 
         outputBuffer[0] = f(dpa.p());                               // Expect: 6
         outputBuffer[1] = f(dpf3.p());                              // Expect: 8
-        outputBuffer[2] = __jvp(f)(dpf3).d();                       // Expect: 5.5
-        outputBuffer[3] = __jvp(f)(dpa).d();                        // Expect: 5
-        outputBuffer[4] = __jvp(g)(dpa).d();                        // Expect: 11.0
+        outputBuffer[2] = __fwd_diff(f)(dpf3).d();                       // Expect: 5.5
+        outputBuffer[3] = __fwd_diff(f)(dpa).d();                        // Expect: 5
+        outputBuffer[4] = __fwd_diff(g)(dpa).d();                        // Expect: 11.0
     }
 }
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
     }
diff --git a/tests/autodiff/vector-swizzle-jvp.slang b/tests/autodiff/vector-swizzle-jvp.slang
index 775c0140e..fc726d067 100644
--- a/tests/autodiff/vector-swizzle-jvp.slang
+++ b/tests/autodiff/vector-swizzle-jvp.slang
@@ -27,16 +27,16 @@ void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID)
         float3 a = float3(2.0, 2.0, 2.0);
         float3 da = float3(1.0, 0.5, 1.0);
 
-        outputBuffer[0] = __jvp(f)(dpfloat3(a, da)).d().x;  // Expect: 1
-        outputBuffer[1] = __jvp(f)(dpfloat3(a, da)).d().y;  // Expect: 0.5
+        outputBuffer[0] = __fwd_diff(f)(dpfloat3(a, da)).d().x;  // Expect: 1
+        outputBuffer[1] = __fwd_diff(f)(dpfloat3(a, da)).d().y;  // Expect: 0.5
         
         float3 x = float3(0.5, 2.0, 0.5);
         float4 y = float4(-1.5, 1.0, 4.0, 2.0);
         float3 dx = float3(1.0, 0.0, -1.0);
         float4 dy = float4(0.0, 0.5, -0.25, 1.0);
 
-        outputBuffer[2] = __jvp(g)(dpfloat3(x, dx), dpfloat4(y, dy)).d().x; // Expect: -2.25
-        outputBuffer[3] = __jvp(g)(dpfloat3(x, dx), dpfloat4(y, dy)).d().y; // Expect: 0.5
+        outputBuffer[2] = __fwd_diff(g)(dpfloat3(x, dx), dpfloat4(y, dy)).d().x; // Expect: -2.25
+        outputBuffer[3] = __fwd_diff(g)(dpfloat3(x, dx), dpfloat4(y, dy)).d().y; // Expect: 0.5
         
     }
 }