Skip to content

SpecialFunctions meta-issue #381

@wsmoses

Description

@wsmoses

Each of the following functions from the SpecialFunctions.jl package should have a method added here that handles TracedRNumber or TracedRArray inputs (depending on whether the expected input is a scalar or vector). This should be implemented using corresponding calls to stablehlo methods https://github.com/openxla/stablehlo/blob/c2fccec9f02daa2d709001ed5f37924d2526676f/stablehlo/dialect/StablehloOps.td#L938 or chlo methods https://github.com/openxla/stablehlo/blob/c2fccec9f02daa2d709001ed5f37924d2526676f/stablehlo/dialect/ChloOps.td . Ideally it should directly call operations within Ops.jl which wrap these stablehlo and chlo utilities. If one needs to call a stablehlo or chlo method which doesn't have an interface in Ops, it should be added to Ops, with a corresponding test, and then use that new Ops.jl method.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions