Skip to content

GPU adaptive step size #53

@samtalki

Description

@samtalki

AdaptiveStepSize currently throws on _GPUStyle() paths (step_rule
dispatch in src/solver.jl). per-problem backtracking does scalar
comparisons that aren't GPU-friendly without rework.

  • per-problem backtracking via masked broadcast over the active subset
  • device-resident Vector{Bool} mask reused across iterations
  • verify the lipschitz-estimate update doesn't require scalar comparison
  • device test under MARGUERITE_TEST_GROUP=gpu

cross-vendor: should work uniformly on Metal / CUDA / AMDGPU once the
broadcast pattern is right.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions