[TreatAsDifferentiable] functions. (#2720)

2 files changed, 39 insertions, 0 deletions

diff --git a/tests/autodiff/treat-as-differentiable.slang b/tests/autodiff/treat-as-differentiable.slang
new file mode 100644
index 000000000..95423d978
--- /dev/null
+++ b/tests/autodiff/treat-as-differentiable.slang
@@ -0,0 +1,37 @@
+// Tests automatic synthesis of Differential type and method requirements.
+
+//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type
+//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type
+
+//TEST_INPUT:ubuffer(data=[0], stride=4):out,name=outputBuffer
+RWStructuredBuffer<float> outputBuffer;
+
+interface IFoo
+{
+    [BackwardDifferentiable]
+    float f(float v);
+}
+
+struct B : IFoo
+{
+    [TreatAsDifferentiable]
+    float f(float v)
+    {
+        return v * v;
+    }
+}
+
+[BackwardDifferentiable]
+float use(IFoo o, float x)
+{
+    return o.f(x);
+}
+
+[numthreads(1, 1, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    B b;
+    var p = diffPair(1.0);
+    __bwd_diff(use)(b, p, 1.0);
+    outputBuffer[0] = p.d;
+}
diff --git a/tests/autodiff/treat-as-differentiable.slang.expected.txt b/tests/autodiff/treat-as-differentiable.slang.expected.txt
new file mode 100644
index 000000000..9d11e5c94
--- /dev/null
+++ b/tests/autodiff/treat-as-differentiable.slang.expected.txt
@@ -0,0 +1,2 @@
+type: float
+0.0
\ No newline at end of file