Skip to content

GeneralizedProximalGradient not working with CuPy #240

@mrava87

Description

@mrava87

Problem

GeneralizedProximalGradient is currently broken with CuPy when show=True because `x[0]/x[0, 0] are not moved to numpy:

x[0] if x.ndim == 1 else x[0, 0],

Solution

Use np.real(to_numpy(x[0])) if x.ndim == 1 else np.real(to_numpy(x[0, 0])) instead.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions