From 79af29af91fb9601886d539526a4ec87bca3d74c Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Thu, 27 Oct 2022 13:08:30 -0700
Subject: Rename `[__custom_jvp]` -> `[ForwardDerivative]`. (#2473)

* Rename `[__custom_jvp]` -> `[ForwardDerivative]`.

* Rename the classes.

* More renaming.

Co-authored-by: Yong He <yhe@nvidia.com>
---
 tests/autodiff/arithmetic-jvp.slang          | 2 +-
 tests/autodiff/custom-intrinsic.slang        | 8 ++++----
 tests/autodiff/generic-custom-jvp.slang      | 2 +-
 tests/autodiff/generic-impl-jvp.slang        | 2 +-
 tests/autodiff/local-redecl-custom-jvp.slang | 2 +-
 tests/autodiff/nested-jvp.slang              | 4 ++--
 tests/autodiff/test-intrinsics-jvp.slang     | 4 ++--
 7 files changed, 12 insertions(+), 12 deletions(-)

(limited to 'tests')

diff --git a/tests/autodiff/arithmetic-jvp.slang b/tests/autodiff/arithmetic-jvp.slang
index 3b06393d3..ec2c5bc6f 100644
--- a/tests/autodiff/arithmetic-jvp.slang
+++ b/tests/autodiff/arithmetic-jvp.slang
@@ -18,7 +18,7 @@ dpfloat g_jvp_(dpfloat dpx)
     return dpfloat(dpx.p(), 2 * dpx.d());
 }
 
-[__custom_jvp(g_jvp_)]
+[ForwardDerivative(g_jvp_)]
 float g(float x)
 {
     return x + x;
diff --git a/tests/autodiff/custom-intrinsic.slang b/tests/autodiff/custom-intrinsic.slang
index 770a18b1e..7591cd624 100644
--- a/tests/autodiff/custom-intrinsic.slang
+++ b/tests/autodiff/custom-intrinsic.slang
@@ -16,7 +16,7 @@ namespace myintrinsiclib
     __target_intrinsic(cuda, "$P_exp($0)")
     __target_intrinsic(cpp, "$P_exp($0)")
     __target_intrinsic(spirv_direct, "12 resultType resultId glsl450 27 _0")
-    [__custom_jvp(d_exp<T>)]
+    [ForwardDerivative(d_exp<T>)]
     T exp(T x);
 
     __generic<T : IDFloat>
@@ -35,7 +35,7 @@ namespace myintrinsiclib
     __target_intrinsic(cuda, "$P_sin($0)")
     __target_intrinsic(cpp, "$P_sin($0)")
     __target_intrinsic(spirv_direct, "12 resultType resultId glsl450 13 _0")
-    [__custom_jvp(d_sin<T>)]
+    [ForwardDerivative(d_sin<T>)]
     T sin(T x);
 
     __generic<T : IDFloat>
@@ -53,7 +53,7 @@ namespace myintrinsiclib
     __target_intrinsic(cuda, "$P_cos($0)")
     __target_intrinsic(cpp, "$P_cos($0)")
     __target_intrinsic(spirv_direct, "12 resultType resultId glsl450 14 _0")
-    [__custom_jvp(d_cos<T>)]
+    [ForwardDerivative(d_cos<T>)]
     T cos(T x);
 
     __generic<T : IDFloat>
@@ -68,7 +68,7 @@ namespace myintrinsiclib
     __generic<T : IDFloat>
     __target_intrinsic(hlsl)
     __target_intrinsic(cuda, "$P_sincos($0, $1, $2)")
-    [__custom_jvp(d_sincos<T>)]
+    [ForwardDerivative(d_sincos<T>)]
     void sincos(T x, out T s, out T c)
     {
         s = sin(x);
diff --git a/tests/autodiff/generic-custom-jvp.slang b/tests/autodiff/generic-custom-jvp.slang
index f0b8d3898..5111f0e48 100644
--- a/tests/autodiff/generic-custom-jvp.slang
+++ b/tests/autodiff/generic-custom-jvp.slang
@@ -17,7 +17,7 @@ dpfloat my_pow_jvp(dpfloat x, dpfloat n)
         x.d() * n.p() * pow(x.p(), n.p()-1) + n.d() * pow(x.p(), n.p()) * log(x.p()));
 }
 
-[__custom_jvp(my_pow_jvp)]
+[ForwardDerivative(my_pow_jvp)]
 float _pow(float, float);
 
 [numthreads(1, 1, 1)]
diff --git a/tests/autodiff/generic-impl-jvp.slang b/tests/autodiff/generic-impl-jvp.slang
index 3ebbff996..d47da336e 100644
--- a/tests/autodiff/generic-impl-jvp.slang
+++ b/tests/autodiff/generic-impl-jvp.slang
@@ -105,7 +105,7 @@ myvector<T, N> operator *(T a, myvector<T, N> b)
 }
 
 __generic<T : IDFloat, let N : int>
-[__custom_jvp(dot_jvp)]
+[ForwardDerivative(dot_jvp)]
 T dot(myvector<T, N> a, myvector<T, N> b)
 {
     T curr = (T)0.0;
diff --git a/tests/autodiff/local-redecl-custom-jvp.slang b/tests/autodiff/local-redecl-custom-jvp.slang
index 79b90bd16..3a6b6f474 100644
--- a/tests/autodiff/local-redecl-custom-jvp.slang
+++ b/tests/autodiff/local-redecl-custom-jvp.slang
@@ -15,7 +15,7 @@ dpfloat my_pow_jvp(dpfloat x, dpfloat n)
         x.d() * n.p() * pow(x.p(), n.p()-1) + n.d() * pow(x.p(), n.p()) * log(x.p()));
 }
 
-[__custom_jvp(my_pow_jvp)]
+[ForwardDerivative(my_pow_jvp)]
 float _pow(float, float);
 
 [numthreads(1, 1, 1)]
diff --git a/tests/autodiff/nested-jvp.slang b/tests/autodiff/nested-jvp.slang
index 40518d44d..0e7d19078 100644
--- a/tests/autodiff/nested-jvp.slang
+++ b/tests/autodiff/nested-jvp.slang
@@ -7,13 +7,13 @@ RWStructuredBuffer<float> outputBuffer;
 typedef __DifferentialPair<float> dpfloat;
 typedef __DifferentialPair<float3> dpfloat3;
 
-[__custom_jvp(pow_jvp)]
+[ForwardDerivative(pow_jvp)]
 float pow_(float x, float n)
 {
     return pow<float>(x, n);
 }
 
-[__custom_jvp(max_jvp)]
+[ForwardDerivative(max_jvp)]
 float max_(float x, float y)
 {
     return max<float>(x, y);
diff --git a/tests/autodiff/test-intrinsics-jvp.slang b/tests/autodiff/test-intrinsics-jvp.slang
index cb4c5c6b4..39f2ee495 100644
--- a/tests/autodiff/test-intrinsics-jvp.slang
+++ b/tests/autodiff/test-intrinsics-jvp.slang
@@ -2,14 +2,14 @@
 
 __exported import test_intrinsics;
 
-[__custom_jvp(pow_jvp)]
+[ForwardDerivative(pow_jvp)]
 float pow_(float x, float n);
 float pow_jvp(float x, float n, float dx, float dn)
 {
     return dx * n * pow(x, n-1) + dn * pow(x, n) * log(x);
 }
 
-[__custom_jvp(max_jvp)]
+[ForwardDerivative(max_jvp)]
 float max_(float x, float y);
 float max_jvp(float x, float y, float dx, float dy)
 {
-- 
cgit v1.2.3