提交 e10b9515 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Don't run useless fusion and inplace rewrites in JAX mode

上级 36161e80
......@@ -454,7 +454,10 @@ else:
JAX = Mode(
JAXLinker(),
RewriteDatabaseQuery(include=["fast_run", "jax"], exclude=["cxx_only", "BlasOpt"]),
RewriteDatabaseQuery(
include=["fast_run", "jax"],
exclude=["cxx_only", "BlasOpt", "fusion", "inplace"],
),
)
NUMBA = Mode(
NumbaLinker(),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论