提交 d016564b authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Numba linalg: Fix obj fallback raise -> return

上级 d64cceaf
......@@ -246,7 +246,7 @@ def numba_funcify_Solve(op, node, **kwargs):
out_dtype = node.outputs[0].type.numpy_dtype
if A_dtype.kind == "c" or b_dtype.kind == "c":
raise generate_fallback_impl(op, node=node, **kwargs)
return generate_fallback_impl(op, node=node, **kwargs)
must_cast_A = A_dtype != out_dtype
if must_cast_A and config.compiler_verbose:
print("Solve requires casting first input `A`") # noqa: T201
......@@ -320,7 +320,7 @@ def numba_funcify_SolveTriangular(op, node, **kwargs):
out_dtype = node.outputs[0].type.numpy_dtype
if A_dtype.kind == "c" or b_dtype.kind == "c":
raise generate_fallback_impl(op, node=node, **kwargs)
return generate_fallback_impl(op, node=node, **kwargs)
must_cast_A = A_dtype != out_dtype
if must_cast_A and config.compiler_verbose:
print("SolveTriangular requires casting first input `A`") # noqa: T201
......@@ -371,7 +371,7 @@ def numba_funcify_CholeskySolve(op, node, **kwargs):
out_dtype = node.outputs[0].type.numpy_dtype
if c_dtype.kind == "c" or b_dtype.kind == "c":
raise generate_fallback_impl(op, node=node, **kwargs)
return generate_fallback_impl(op, node=node, **kwargs)
must_cast_c = c_dtype != out_dtype
if must_cast_c and config.compiler_verbose:
print("CholeskySolve requires casting first input `c`") # noqa: T201
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论