From a670bafc121c20168624f70a388dbe8556402c7f Mon Sep 17 00:00:00 2001 From: kaizhangNV <149626564+kaizhangNV@users.noreply.github.com> Date: Wed, 9 Jul 2025 11:25:29 -0500 Subject: no_diff diagnostics improvement (#7655) close #6286. This PR is to improve the diagnostics for no_diff usage. In a differentiable function, any calls to a non-diff function with constant arguments should not require no_diff attribute. This PR adds this extra check at `checkAutoDiffUsages` where it checks the differentiability on IR. In a differentiable method, we will force to use `[NoDiffThis]` attribute if there is access to non-differentiable `This` type. Once this access is detected we will report a warning to bring users attention that this access won't generate any derivative, they have to use `[NoDiffThis]` to suppress that warning. This PR adds this check at type checking stage, because it's the easiest way to find out all the `This` accesses. --- tests/diagnostics/force-no-diff-this.slang | 42 ++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 tests/diagnostics/force-no-diff-this.slang (limited to 'tests/diagnostics/force-no-diff-this.slang') 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 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 s = {x}; + return s.eval(x) + s.eval1(x); +} + +RWStructuredBuffer 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; +} + -- cgit v1.2.3