3 files changed, 6 insertions, 0 deletions
diff --git a/tests/autodiff/generic-impl-jvp.slang b/tests/autodiff/generic-impl-jvp.slang
index 674d5c5ca..4e28eeaf8 100644
--- a/tests/autodiff/generic-impl-jvp.slang
+++ b/tests/autodiff/generic-impl-jvp.slang
@@ -11,6 +11,7 @@ typealias IDFloat = __BuiltinRealType & IDifferentiable;
 __generic<T : IDifferentiable, let N : int>
 struct dvector : IDifferentiable
 {
+    typedef dvector<T.Differential, N> Differential;
     T values[N];
 };
 
@@ -147,6 +148,8 @@ DifferentialPair<T> dot_jvp(dpvector<T, N> a, dpvector<T, N> b)
 __generic<let N : int>
 struct lineardvector : IDifferentiable
 {
+    typedef lineardvector<N> Differential;
+
     myvector<Real, N>.Differential val;
 
     __init(vector<Real.Differential, N> a)
diff --git a/tests/autodiff/getter-setter-multi.slang b/tests/autodiff/getter-setter-multi.slang
index 9f03ac4eb..3f8257897 100644
--- a/tests/autodiff/getter-setter-multi.slang
+++ b/tests/autodiff/getter-setter-multi.slang
@@ -6,6 +6,7 @@ RWStructuredBuffer<float> outputBuffer;
 
 struct B : IDifferentiable
 {
+    typedef B Differential;
     float3 z;
     float.Differential k[10];
 };
diff --git a/tests/autodiff/getter-setter.slang b/tests/autodiff/getter-setter.slang
index bc7343f27..1b0b8232e 100644
--- a/tests/autodiff/getter-setter.slang
+++ b/tests/autodiff/getter-setter.slang
@@ -6,6 +6,8 @@ RWStructuredBuffer<float> outputBuffer;
 
 struct B : IDifferentiable
 {
+    typedef B Differential;
+
     float z;
 };