Skip to content

Commit

Permalink
fixing dpnp failure caused by addition of nin, nout and ntypes
Browse files Browse the repository at this point in the history
  • Loading branch information
adarshyoga committed Apr 10, 2024
1 parent 1336f76 commit 207e90f
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions numba_dpex/dpnp_iface/dpnp_ufunc_db.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,20 +72,23 @@ def _fill_ufunc_db_with_dpnp_ufuncs(ufunc_db):
"d->d": mathimpl.lower_ocl_impl[("erf", (_unary_d_d))],
}
else:
op = getattr(dpnp, ufuncop)
dpnpop = getattr(dpnp, ufuncop)
npop = getattr(np, ufuncop)
op.nin = npop.nin
op.nout = npop.nout
op.nargs = npop.nargs
op.types = npop.types
op.is_dpnp_ufunc = True
if not hasattr(dpnpop, "nin"):
dpnpop.nin = npop.nin
if not hasattr(dpnpop, "nout"):
dpnpop.nout = npop.nout
dpnpop.nargs = dpnpop.nin + dpnpop.nout
if not hasattr(dpnpop, "types"):
dpnpop.types = npop.types
dpnpop.is_dpnp_ufunc = True
cp = copy.copy(_ufunc_db[npop])
ufunc_db.update({op: cp})
for key in list(ufunc_db[op].keys()):
ufunc_db.update({dpnpop: cp})
for key in list(ufunc_db[dpnpop].keys()):
if (
"FF->" in key
or "DD->" in key
or "F->" in key
or "D->" in key
):
ufunc_db[op].pop(key)
ufunc_db[dpnpop].pop(key)

0 comments on commit 207e90f

Please sign in to comment.