diff options
Diffstat (limited to 'tests')
11 files changed, 144 insertions, 0 deletions
diff --git a/tests/autodiff/dynamic-dispatch-bwd-diff.slang b/tests/autodiff/dynamic-dispatch-bwd-diff.slang index 5945c22cd..9941aa7b1 100644 --- a/tests/autodiff/dynamic-dispatch-bwd-diff.slang +++ b/tests/autodiff/dynamic-dispatch-bwd-diff.slang @@ -17,6 +17,7 @@ struct A : IInterface { float a; [BackwardDifferentiable] + [NoDiffThis] float calc(float x) { return a*x*x; } }; @@ -24,6 +25,7 @@ struct B : IInterface { float a; [BackwardDifferentiable] + [NoDiffThis] float calc(float x) { return a*x*x*x; } }; diff --git a/tests/autodiff/dynamic-dispatch-generic-2.slang b/tests/autodiff/dynamic-dispatch-generic-2.slang index bbf7c7da1..97cb5f42c 100644 --- a/tests/autodiff/dynamic-dispatch-generic-2.slang +++ b/tests/autodiff/dynamic-dispatch-generic-2.slang @@ -16,7 +16,9 @@ interface IInterface struct A : IInterface { float z; + [ForwardDifferentiable] + [NoDiffThis] float calc(float x) { return x * x * x; } }; @@ -25,6 +27,7 @@ struct B : IInterface float z; [ForwardDifferentiable] + [NoDiffThis] float calc(float x) { return x * x + z; } }; diff --git a/tests/autodiff/dynamic-object-bwd-diff.slang b/tests/autodiff/dynamic-object-bwd-diff.slang index a10c48f9b..a80025d52 100644 --- a/tests/autodiff/dynamic-object-bwd-diff.slang +++ b/tests/autodiff/dynamic-object-bwd-diff.slang @@ -26,7 +26,9 @@ struct C : IInterface2 struct A : IInterface { float a; + [BackwardDifferentiable] + [NoDiffThis] float calc(IInterface2 i2, float x) { float b = no_diff(i2.innerCalc(x)); @@ -37,7 +39,9 @@ struct A : IInterface struct B : IInterface { float a; + [BackwardDifferentiable] + [NoDiffThis] float calc(IInterface2 i2, float x) { float b = no_diff(i2.innerCalc(x)); diff --git a/tests/autodiff/material2/DiffuseMaterial.slang b/tests/autodiff/material2/DiffuseMaterial.slang index 1422ee30c..a5c0aaa30 100644 --- a/tests/autodiff/material2/DiffuseMaterial.slang +++ b/tests/autodiff/material2/DiffuseMaterial.slang @@ -16,6 +16,7 @@ public struct DiffuseMaterial : IMaterial } [BackwardDifferentiable] + [NoDiffThis] public DiffuseMaterialInstance setupMaterialInstance(out MaterialInstanceData miData) { float3 albedo = getAlbedo(baseColor); diff --git a/tests/autodiff/material2/GlossyMaterial.slang b/tests/autodiff/material2/GlossyMaterial.slang index 1070c6e63..12fbc5f73 100644 --- a/tests/autodiff/material2/GlossyMaterial.slang +++ b/tests/autodiff/material2/GlossyMaterial.slang @@ -30,6 +30,7 @@ public struct GlossyMaterial : IMaterial } [BackwardDifferentiable] + [NoDiffThis] public GlossyMaterialInstance setupMaterialInstance(out MaterialInstanceData miData) { float3 albedo = getAlbedo(baseColor); diff --git a/tests/autodiff/material2/MxLayeredMaterial.slang b/tests/autodiff/material2/MxLayeredMaterial.slang index fabfde80c..9e88d2ce9 100644 --- a/tests/autodiff/material2/MxLayeredMaterial.slang +++ b/tests/autodiff/material2/MxLayeredMaterial.slang @@ -52,6 +52,7 @@ public struct MxLayeredMaterial : IMaterial } [Differentiable] + [NoDiffThis] public UsedMaterialInstance setupMaterialInstance(out MaterialInstanceData miData) { float3 albedo = getAlbedo(baseColor); diff --git a/tests/autodiff/member-func-extension-custom-derivative.slang b/tests/autodiff/member-func-extension-custom-derivative.slang index 8752dfff5..72fbcf9a1 100644 --- a/tests/autodiff/member-func-extension-custom-derivative.slang +++ b/tests/autodiff/member-func-extension-custom-derivative.slang @@ -32,6 +32,7 @@ struct Foo<T : IDifferentiable> extension<T : IDifferentiable> Foo<T> { [ForwardDerivativeOf(doThing)] + [NoDiffThis] DifferentialPair<T> diff_doThing() { return diffPair(value, T.dzero()); diff --git a/tests/autodiff/nondiff-call.slang b/tests/autodiff/nondiff-call.slang index d62de1b78..79d9b9f15 100644 --- a/tests/autodiff/nondiff-call.slang +++ b/tests/autodiff/nondiff-call.slang @@ -35,6 +35,7 @@ struct A float o; [ForwardDifferentiable] + [NoDiffThis] float doSomethingDifferentiable(float b) { return o + b; diff --git a/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement.slang b/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement.slang new file mode 100644 index 000000000..961ac75d5 --- /dev/null +++ b/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement.slang @@ -0,0 +1,40 @@ +//DIAGNOSTIC_TEST:SIMPLE(filecheck=CHECK): + +float someNoDiffFunc(float x, no_diff float y) +{ + return x * x + y * y; +} + +// Previously, when we call a no-diff function side a differntiable function, we will have to use no_diff to tell compiler that this is intended. +// However, if the parameter is just a constant, there is no need to use no_diff, because constant won't carry any derivative information. +// Therefore, this test is to check we won't report any error when the parameter is a constant in this case. +[Differentiable] +float eval(float x) +{ + // CHECK-NOT: ([[# @LINE+1]]): error 41020 + return exp(x) - someNoDiffFunc(1.0f, x); +} + +[Differentiable] +float eval1(float x) +{ + // CHECK: ([[# @LINE+1]]): error 41020 + return exp(x) - someNoDiffFunc(x, 1.0); +} + +RWStructuredBuffer<float> output; + +[shader("compute")] +[numthreads(1,1,1)] +void computeMain(uint id : SV_DispatchThreadID) +{ + var x = diffPair(2.0f); + bwd_diff(eval)(x, 1.0f); + + output[0] = x.d; + + var x1 = diffPair(2.0f); + bwd_diff(eval1)(x1, 1.0f); + output[1] = x1.d; +} + diff --git a/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement1.slang b/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement1.slang new file mode 100644 index 000000000..f27c6ec6b --- /dev/null +++ b/tests/diagnostics/const-to-nodiff-function-diagnostic-improvement1.slang @@ -0,0 +1,48 @@ +//DIAGNOSTIC_TEST:SIMPLE(filecheck=CHECK): + + +// Similar to const-to-nodiff-function-diagnostic-improvement.slang, but with a CoopVec type +// to reproduce a more realistic scenario. +extension<T : __BuiltinFloatingPointType, let K : int> CoopVec<T, K> : IDifferentiable +{ + typealias Differential = CoopVec<T, K>; +}; + +[BackwardDerivativeOf(exp)] +void exp_BackwardAutoDiff<T : __BuiltinFloatingPointType, let K : int>(inout DifferentialPair<CoopVec<T, K>> p0, CoopVec<T, K>.Differential dResult) +{ + p0 = diffPair(p0.p, dResult * exp(p0.p)); +} + +[Differentiable] +CoopVec<T, K> eval<T : __BuiltinFloatingPointType, let K : int>(CoopVec<T, K> x) +{ + // CHECK-NOT: ([[# @LINE+1]]): error 41020 + return exp(x) - CoopVec<T, K>(1.); +} + +[Differentiable] +CoopVec<T, K> eval1<T : __BuiltinFloatingPointType, let K : int>(CoopVec<T, K> x) +{ + // test.slang(25): error 41020: derivative cannot be propagated through call to non-backward-differentiable function `CoopVec.$init`, use 'no_diff' to clarify intention. + // CHECK: ([[# @LINE+1]]): error 41020 + return exp(x) - CoopVec<T, K>(x[0]); +} + + +RWStructuredBuffer<float> output; + +[shader("compute")] +[numthreads(1,1,1)] +void computeMain(uint id : SV_DispatchThreadID) +{ + var x = diffPair(CoopVec<float, 2>(2.0f), CoopVec<float, 2>(1.0f)); + bwd_diff(eval)(x, CoopVec<float, 2>(1.0f)); + + output[0] = x.d[0]; + + var x1 = diffPair(CoopVec<float, 2>(2.0f), CoopVec<float, 2>(1.0f)); + bwd_diff(eval1)(x1, CoopVec<float, 2>(1.0f)); + output[1] = x1.d[1]; +} + diff --git a/tests/diagnostics/force-no-diff-this.slang b/tests/diagnostics/force-no-diff-this.slang new file mode 100644 index 000000000..ae1464ffb --- /dev/null +++ b/tests/diagnostics/force-no-diff-this.slang @@ -0,0 +1,42 @@ +//DIAGNOSTIC_TEST:SIMPLE(filecheck=CHECK): + +struct MyStruct<T> where T: __BuiltinFloatingPointType +{ + float a; + __init(float a) { this.a = a;} + + [Differentiable] + T eval(T x) + { + //CHECK: ([[# @LINE+1]]): warning 31159 + return exp(x * T(a) * T(a)); + } + + [Differentiable] + [NoDiffThis] + T eval1(T x) + { + //CHECK-NOT: ([[# @LINE+1]]): warning 31159 + return exp(x * T(a) * T(a)); + } +}; + +[Differentiable] +float evalFunc(float x) +{ + MyStruct<float> s = {x}; + return s.eval(x) + s.eval1(x); +} + +RWStructuredBuffer<float> output; + +[shader("compute")] +[numthreads(1,1,1)] +void computeMain(uint id : SV_DispatchThreadID) +{ + var x = diffPair(2.0f); + bwd_diff(evalFunc)(x, 1.0f); + + output[0] = x.d; +} + |