Raise Mooncake compat floor to 0.5.36 (fix Downgrade resolution)

.github/dependabot.yml

@@ -5,6 +5,12 @@ updates:
     directory: "/" # Location of package manifests
     schedule:
       interval: "weekly"
-    ignore:
-      - dependency-name: "crate-ci/typos"
-        update-types: ["version-update:semver-patch", "version-update:semver-minor"]
+  - package-ecosystem: "julia"
+    directories:
+      - "/"
+    schedule:
+      interval: "daily"
+    groups:
+      all-julia-packages:
+        patterns:
+          - "*"

.github/workflows/DependabotAutoMerge.yml

@@ -0,0 +1,9 @@
+name: "Dependabot Auto-merge"
+on: pull_request
+permissions:
+  contents: write
+  pull-requests: write
+jobs:
+  automerge:
+    uses: "SciML/.github/.github/workflows/dependabot-automerge.yml@v1"
+    secrets: "inherit"

.github/workflows/Downgrade.yml

@@ -12,20 +12,7 @@ on:
       - 'docs/**'
 jobs:
   test:
-    runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        downgrade_mode: ['alldeps']
-        julia-version: ['1.10']
-    steps:
-      - uses: actions/checkout@v4
-      - uses: julia-actions/setup-julia@v2
-        with:
-          version: ${{ matrix.julia-version }}
-      - uses: julia-actions/julia-downgrade-compat@v2
-        with:
-          skip: Pkg,TOML
-      - uses: julia-actions/julia-buildpkg@v1
-      - uses: julia-actions/julia-runtest@v1
-        with:
-          ALLOW_RERESOLVE: false
+    uses: "SciML/.github/.github/workflows/downgrade.yml@v1"
+    with:
+      julia-version: "lts"
+    secrets: "inherit"

.github/workflows/Downstream.yml

@@ -12,14 +12,10 @@ concurrency:
 jobs:
   test:
     name: ${{ matrix.package.repo }}/${{ matrix.package.group }}/${{ matrix.julia-version }}
-    runs-on: ${{ matrix.os }}
-    env:
-      GROUP: ${{ matrix.package.group }}
     strategy:
       fail-fast: false
       matrix:
-        julia-version: [1]
-        os: [ubuntu-latest]
+        julia-version: ["1"]
         package:
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: InterfaceI}
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: InterfaceII}
@@ -31,42 +27,11 @@ jobs:
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: RegressionI}
           - {user: SciML, repo: OrdinaryDiffEq.jl, group: RegressionII}
           - {user: SciML, repo: SimpleDiffEq.jl, group: Core}
-
-    steps:
-      - uses: actions/checkout@v4
-      - uses: julia-actions/setup-julia@v2
-        with:
-          version: ${{ matrix.julia-version }}
-          arch: x64
-      - uses: julia-actions/julia-buildpkg@latest
-      - name: Clone Downstream
-        uses: actions/checkout@v4
-        with:
-          repository: ${{ matrix.package.user }}/${{ matrix.package.repo }}
-          path: downstream
-      - name: Load this and run the downstream tests
-        shell: julia --color=yes --project=downstream {0}
-        run: |
-          using Pkg
-          try
-            # force it to use this PR's version of the package
-            Pkg.develop(PackageSpec(path="."))  # resolver may fail with main deps
-            Pkg.update()
-            Pkg.test(coverage=true)  # resolver may fail with test time deps
-          catch err
-            err isa Pkg.Resolve.ResolverError || rethrow()
-            # If we can't resolve that means this is incompatible by SemVer and this is fine
-            # It means we marked this as a breaking change, so we don't need to worry about
-            # Mistakenly introducing a breaking change, as we have intentionally made one
-            @info "Not compatible with this release. No problem." exception=err
-            exit(0)  # Exit immediately, as a success
-          end
-        env:
-          RETESTITEMS_NWORKERS: 4
-          RETESTITEMS_NWORKER_THREADS: 2
-      - uses: julia-actions/julia-processcoverage@v1
-      - uses: codecov/codecov-action@v5
-        with:
-          files: lcov.info
-          token: ${{ secrets.CODECOV_TOKEN }}
-          fail_ci_if_error: true
+    uses: "SciML/.github/.github/workflows/downstream.yml@v1"
+    with:
+      julia-version: "${{ matrix.julia-version }}"
+      julia-arch: "x64"
+      owner: "${{ matrix.package.user }}"
+      repo: "${{ matrix.package.repo }}"
+      group: "${{ matrix.package.group }}"
+    secrets: "inherit"

.github/workflows/FormatCheck.yml

@@ -1,13 +1,18 @@
-name: "Format Check"
+name: format-check
 
 on:
   push:
     branches:
+      - 'master'
       - 'main'
+      - 'release-'
     tags: '*'
   pull_request:
 
 jobs:
-  format-check:
-    name: "Format Check"
-    uses: "SciML/.github/.github/workflows/format-check.yml@v1"
+  runic:
+    uses: "SciML/.github/.github/workflows/runic.yml@v1"
+    with:
+      julia-version: "1"
+      runic-version: "1"
+    secrets: "inherit"

.github/workflows/RunicSuggestions.yml

@@ -0,0 +1,9 @@
+name: "Runic Suggestions"
+on: pull_request
+permissions:
+  contents: read
+  pull-requests: write
+jobs:
+  runic-suggestions:
+    uses: "SciML/.github/.github/workflows/runic-suggestions-on-pr.yml@v1"
+    secrets: "inherit"

.github/workflows/SpellCheck.yml

@@ -5,9 +5,5 @@ on: [pull_request]
 jobs:
   typos-check:
     name: Spell Check with Typos
-    runs-on: ubuntu-latest
-    steps:
-      - name: Checkout Actions Repository
-        uses: actions/checkout@v4
-      - name: Check spelling
-        uses: crate-ci/typos@v1.18.0
+    uses: "SciML/.github/.github/workflows/spellcheck.yml@v1"
+    secrets: "inherit"

.github/workflows/TagBot.yml

@@ -1,15 +1,9 @@
-name: TagBot
+name: "TagBot"
 on:
   issue_comment:
-    types:
-      - created
+    types: [created]
   workflow_dispatch:
 jobs:
-  TagBot:
-    if: github.event_name == 'workflow_dispatch' || github.actor == 'JuliaTagBot'
-    runs-on: ubuntu-latest
-    steps:
-      - uses: JuliaRegistries/TagBot@v1
-        with:
-          token: ${{ secrets.GITHUB_TOKEN }}
-          ssh: ${{ secrets.DOCUMENTER_KEY }}
+  tagbot:
+    uses: "SciML/.github/.github/workflows/tagbot.yml@v1"
+    secrets: "inherit"

.github/workflows/Tests.yml

@@ -14,18 +14,5 @@ concurrency:
 
 jobs:
   tests:
-    name: "Tests"
-    strategy:
-      fail-fast: false
-      matrix:
-        version:
-          - "1"
-          - "lts"
-          - "pre"
-        group:
-          - Core
-    uses: "SciML/.github/.github/workflows/tests.yml@v1"
-    with:
-      julia-version: "${{ matrix.version }}"
-      group: "${{ matrix.group }}"
+    uses: "SciML/.github/.github/workflows/grouped-tests.yml@v1"
     secrets: "inherit"

Project.toml

@@ -1,7 +1,7 @@
 name = "FastPower"
 uuid = "a4df4552-cc26-4903-aec0-212e50a0e84b"
 authors = ["Chris Rackauckas <accounts@chrisrackauckas.com>"]
-version = "1.1.3"
+version = "1.3.4"
 
 [deps]
 
@@ -24,12 +24,17 @@ FastPowerReverseDiffExt = "ReverseDiff"
 FastPowerTrackerExt = "Tracker"
 
 [compat]
-Enzyme = "0.13"
+Enzyme = "0.13.89"
+EnzymeTestUtils = "0.2"
 ForwardDiff = "0.10, 1"
-Measurements = "2"
+Measurements = "2.5"
 MonteCarloMeasurements = "1"
-Mooncake = "0.4"
-ReverseDiff = "1"
+Mooncake = "0.5.36"
+ReverseDiff = "1.14"
+SafeTestsets = "0.1, 1"
+SciMLTesting = "1"
+StableRNGs = "1"
+Test = "1"
 Tracker = "0.2"
 julia = "1.10"
 
@@ -39,8 +44,11 @@ EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
-Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
+SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
+SciMLTesting = "09d9d899-5365-40a9-917a-5f67fddea283"
+StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 
 [targets]
-test = ["Test", "Enzyme", "EnzymeTestUtils", "ForwardDiff", "Mooncake", "ReverseDiff", "Tracker"]
+test = ["Enzyme", "EnzymeTestUtils", "ForwardDiff", "Mooncake", "ReverseDiff", "SafeTestsets", "SciMLTesting", "StableRNGs", "Test", "Tracker"]

ext/FastPowerEnzymeExt.jl

@@ -5,53 +5,9 @@ import FastPower: fastpower
 using Enzyme
 using Enzyme.EnzymeRules: FwdConfig
 
-function Enzyme.EnzymeRules.forward(config::FwdConfig,
-        func::Const{typeof(FastPower.fastpower)},
-        RT::Type{<:Union{Duplicated, DuplicatedNoNeed}},
-        _x::Union{Const, Duplicated}, _y::Union{Const, Duplicated})
-    x = _x.val
-    y = _y.val
-    ret = func.val(x, y)
-    T = typeof(ret)
-    if !(_x isa Const)
-        dxval = _x.dval * y * (fastpower(x, y - 1))
-    else
-        dxval = make_zero(_x.val)
-    end
-    if !(_y isa Const)
-        dyval = x isa Real && x<=0 ? Base.oftype(float(x), NaN) :
-                _y.dval*(fastpower(x, y))*log(x)
-    else
-        dyval = make_zero(_y.val)
-    end
-    if RT <: DuplicatedNoNeed
-        return convert(T, dxval + dyval)
-    else
-        return Duplicated(ret, convert(T, dxval + dyval))
-    end
-end
-
-function EnzymeRules.augmented_primal(config::Enzyme.EnzymeRules.RevConfigWidth{1},
-        func::Const{typeof(fastpower)}, ::Union{Type{<:Active}, Type{<:Const}},
-        x::Union{Const, Active}, y::Union{Const, Active})
-    if EnzymeRules.needs_primal(config)
-        primal = func.val(x.val, y.val)
-    else
-        primal = nothing
-    end
-    return EnzymeRules.AugmentedReturn(primal, nothing, nothing)
-end
-
-function EnzymeRules.reverse(config::Enzyme.EnzymeRules.RevConfigWidth{1},
-        func::Const{typeof(fastpower)}, dret, tape, _x::Union{Const, Active}, _y::Union{
-            Const, Active})
-    x = _x.val
-    y = _y.val
-    dxval = _x isa Const ? nothing : dret.val * y * (fastpower(x, y - 1))
-    dyval = _y isa Const ? nothing :
-            (x isa Real && x<=0 ? Base.oftype(float(x), NaN) :
-             dret.val * (fastpower(x, y)) * log(x))
-    return (dxval, dyval)
-end
+Enzyme.EnzymeRules.@easy_rule(
+    FastPower.fastpower(x, y),
+    (y * fastpower(x, y - 1), Ω * log(x))
+)
 
 end

ext/FastPowerMonteCarloMeasurementsExt.jl

@@ -3,9 +3,11 @@ module FastPowerMonteCarloMeasurementsExt
 using FastPower
 using MonteCarloMeasurements
 
-@inline function FastPower.fastpower(x::MonteCarloMeasurements.AbstractParticles,
-        y::MonteCarloMeasurements.AbstractParticles)
-    x^y
+@inline function FastPower.fastpower(
+        x::MonteCarloMeasurements.AbstractParticles,
+        y::MonteCarloMeasurements.AbstractParticles
+    )
+    return x^y
 end
 
 end

test/Enzyme/enzyme_forward_tests.jl

@@ -0,0 +1,28 @@
+using FastPower: fastpower
+using Enzyme, EnzymeTestUtils
+using StableRNGs
+using Test
+
+# `test_forward` compares the rule (which returns the *exact* `^` derivative) against finite
+# differences of the *approximate* `fastpower` primal. Because `fastpower` routes through a
+# Float32 `fastlog2` polynomial, the *slope* of its primal differs from the exact slope by
+# ~1e-2 relative near x=1 (even where the primal value itself is exact), so the FD reference
+# is off from the exact rule by that inherent approximation error rather than by any rule bug.
+# The previous atol=1e-4, rtol=1e-3 sat below that gap, so whether the lane passed depended on
+# the random tangents `test_forward` drew from the global RNG (it went red ~4% of the time).
+#
+# Fix: draw the tangents from a StableRNG (a fixed seed gives the *same* stream on every Julia
+# version, unlike the global RNG / Xoshiro whose stream can change across versions) so the test
+# is genuinely deterministic, and use atol=1e-3, rtol=1e-2 matched to `fastpower`'s documented
+# accuracy envelope (see test/fast_pow_tests.jl). That tolerance is ~5x above the measured
+# worst-case relative discrepancy in this grid (~2e-3) yet far below the O(1) relative error a
+# genuinely wrong derivative rule would produce, so real regressions are still caught. Reverting
+# to rtol=1e-3 is not possible without cherry-picking a lucky seed to hide the inherent gap.
+rng = StableRNG(123)
+@testset for RT in (Duplicated, DuplicatedNoNeed),
+        Tx in (Const, Duplicated),
+        Ty in (Const, Duplicated)
+    x = 1.0
+    y = 0.5
+    test_forward(fastpower, RT, (x, Tx), (y, Ty), rng = rng, atol = 1.0e-3, rtol = 1.0e-2)
+end

test/Enzyme/enzyme_reverse_tests.jl

@@ -0,0 +1,15 @@
+using FastPower: fastpower
+using Enzyme, EnzymeTestUtils
+using StableRNGs
+using Test
+
+# See test/Enzyme/enzyme_forward_tests.jl: the finite-difference reference is taken on the
+# approximate `fastpower` primal, so the tolerance must cover its inherent approximation error.
+# Draw the cotangents from a StableRNG (stream is identical across Julia versions, so the test
+# is deterministic) and match `fastpower`'s documented accuracy with atol=1e-3, rtol=1e-2.
+rng = StableRNG(123)
+@testset for RT in (Active,), Tx in (Active, Const), Ty in (Active, Const)
+    x = 1.0
+    y = 0.5
+    test_reverse(fastpower, RT, (x, Tx), (y, Ty), rng = rng, atol = 1.0e-3, rtol = 1.0e-2)
+end

test/fast_log2_tests.jl

@@ -0,0 +1,6 @@
+using FastPower: fastlog2
+using Test
+
+for x in 0.001:0.001:1.2 # (0, 1+something] is the domain which a controller uses
+    @test log2(x) ≈ fastlog2(Float32(x)) atol = 1.0e-3
+end