From 4ad0470025da4e808c46023f9a2525febcf973a2 Mon Sep 17 00:00:00 2001 From: Yong He Date: Wed, 23 Nov 2022 16:02:56 -0800 Subject: Fix issues around dynamic generic function and autodiff. (#2528) * Fix issues around dynamic generic function and autodiff. * Fix return type issue. * Fix type unification for generic `inout` parameter. * Fix. Co-authored-by: Yong He --- tests/autodiff/dynamic-dispatch-generic.slang | 46 +++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 tests/autodiff/dynamic-dispatch-generic.slang (limited to 'tests/autodiff/dynamic-dispatch-generic.slang') diff --git a/tests/autodiff/dynamic-dispatch-generic.slang b/tests/autodiff/dynamic-dispatch-generic.slang new file mode 100644 index 000000000..37c37a745 --- /dev/null +++ b/tests/autodiff/dynamic-dispatch-generic.slang @@ -0,0 +1,46 @@ +// Test calling differentiable function through dynamic dispatch. + +//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 0 0 0 0], stride=4):out,name=outputBuffer +RWStructuredBuffer outputBuffer; + +[anyValueSize(16)] +interface IInterface +{ + [ForwardDifferentiable] + static float calc(float x); +} + +struct A : IInterface +{ + [ForwardDifferentiable] + static float calc(float x) { return x * x * x; } +}; + +struct B : IInterface +{ + [ForwardDifferentiable] + static float calc(float x) { return x * x; } +}; + +[ForwardDifferentiable] +float sqr(T obj, float x) +{ + return obj.calc(x); +} + +//TEST_INPUT: type_conformance A:IInterface = 0 +//TEST_INPUT: type_conformance B:IInterface = 1 + + +[numthreads(1, 1, 1)] +void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) +{ + var obj = createDynamicObject(dispatchThreadID.x, 0); // A + outputBuffer[0] = __fwd_diff(sqr)(obj, DifferentialPair(2.0, 1.0)).d; // A.calc, expect 12 + + obj = createDynamicObject(dispatchThreadID.x + 1, 0); // B + outputBuffer[1] = __fwd_diff(sqr)(obj, DifferentialPair(1.5, 1.0)).d; // B.calc, expect 3 +} -- cgit v1.2.3