Revision 5902acdabc4445a65741a7a6a3a95f223e301059 authored by Yong He on 23 January 2024, 07:19:40 UTC, committed by GitHub on 23 January 2024, 07:19:40 UTC
Co-authored-by: Yong He <yhe@nvidia.com>
1 parent fec9c42
test-intrinsics-jvp.slang
//TEST_IGNORE_FILE:
__exported import test_intrinsics;
[ForwardDerivative(pow_jvp)]
float pow_(float x, float n);
float pow_jvp(float x, float n, float dx, float dn)
{
return dx * n * pow(x, n-1) + dn * pow(x, n) * log(x);
}
[ForwardDerivative(max_jvp)]
float max_(float x, float y);
float max_jvp(float x, float y, float dx, float dy)
{
return (x > y) ? dx : dy;
}
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...