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

Fix invalid py_file in numba cache locator

上级 bc0d670f
from collections.abc import Callable from collections.abc import Callable
from hashlib import sha256 from hashlib import sha256
from pathlib import Path
from pickle import dump from pickle import dump
from tempfile import NamedTemporaryFile from tempfile import NamedTemporaryFile
from typing import Any from typing import Any
...@@ -64,8 +63,8 @@ class NumbaPyTensorCacheLocator(_CacheLocator): ...@@ -64,8 +63,8 @@ class NumbaPyTensorCacheLocator(_CacheLocator):
@classmethod @classmethod
def from_function(cls, py_func, py_file): def from_function(cls, py_func, py_file):
"""Create a locator instance for functions stored in CACHED_SRC_FUNCTIONS.""" """Create a locator instance for functions stored in CACHED_SRC_FUNCTIONS."""
if config.numba__cache and py_func in CACHED_SRC_FUNCTIONS: if py_func in CACHED_SRC_FUNCTIONS and config.numba__cache:
return cls(py_func, Path(py_file).parent, CACHED_SRC_FUNCTIONS[py_func]) return cls(py_func, py_file, CACHED_SRC_FUNCTIONS[py_func])
# Register our locator at the front of Numba's locator list # Register our locator at the front of Numba's locator list
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论