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