PrimalSubstituteOf attribute causes internal compiler error

[tunabrain]

The following snippet causes an internal compiler error with Slang compilation aborted due to an exception of class Slang::InternalError: assert failure: udfDecor:

struct BufferWithGrad
{
    RWStructuredBuffer<float> primal;
    StructuredBuffer<float> grad;

    [Differentiable]
    void set(float value) { primal[0] = detach(value); }

    [PrimalSubstituteOf(set)]
    void set_subst(float value)
    {
    }

    [BackwardDerivativeOf(set)]
    void set_bwd(inout DifferentialPair<float> d)
    {
        d = diffPair(d.p, grad[0]);
    }
}

BufferWithGrad result;

[Differentiable]
void diffCall()
{
    result.set(0.0f);
}

[shader("compute")]
[numthreads(32, 1, 1)]
void main(uint3 dispatchThreadID: SV_DispatchThreadID)
{
    bwd_diff(diffCall)();
}

[bmillsNV]

@tunabrain is this blocking you right now or is Q1 OK for this issue?

[tunabrain]

If there is a workaround then it's not blocking. If not, then it becomes a tricky issue, since it means we can't compute gradients without invoking side effects from the primal function. Is this a known issue or does this need investigation?