提交 775702f5 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't compile function for the old backend unless required.

上级 313b519d
...@@ -220,7 +220,9 @@ class NanGuardMode(Mode): ...@@ -220,7 +220,9 @@ class NanGuardMode(Mode):
big_is_error = config.NanGuardMode.big_is_error big_is_error = config.NanGuardMode.big_is_error
assert nan_is_error or inf_is_error or big_is_error assert nan_is_error or inf_is_error or big_is_error
compile_gpu_func(nan_is_error, inf_is_error, big_is_error)
if cuda.cuda_enabled:
compile_gpu_func(nan_is_error, inf_is_error, big_is_error)
def do_check_on(value, nd, var=None): def do_check_on(value, nd, var=None):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论