81245414e1d0cbd63c88c59f0b1cfa7666a28614,theano/gpuarray/opt.py,,local_gpu_solve,#Any#Any#Any#Any#,1990

Before Change

        return
    if op.A_structure not in MATRIX_STRUCTURES_SOLVE:
        return
    return GpuCusolverSolve(A_structure=op.A_structure)


@register_inplace()
@local_optimizer([GpuCusolverSolve], inplace=True)

After Change

def local_gpu_solve(op, context_name, inputs, outputs):
    if not cusolver_available:
        return
    if inputs[0].dtype not in ["float16", "float32"]:
        return
    if op.A_structure not in MATRIX_STRUCTURES_SOLVE:
        return
    op = GpuCusolverSolve(A_structure=op.A_structure)
    if inputs[0].dtype == "float16":
        return op(inputs[0].astype("float32"),
                  inputs[1].astype("float32")).astype("float16")
    return op


@register_inplace()
@local_optimizer([GpuCusolverSolve], inplace=True)

Instances