Skip to content

Commit

Permalink
Merge pull request #8154 from leofang/jitify_fix
Browse files Browse the repository at this point in the history
Prevent angular brackets from appearing in Jitify's cache filename
  • Loading branch information
takagi committed Feb 5, 2024
2 parents e2d0b98 + bd32e10 commit 666319b
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion cupy/cuda/jitify.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -226,8 +226,11 @@ cpdef void _init_module() except*:
# start time, for enabling CCCL + CuPy developers?
global _jitify_cache_versions
if _jitify_cache_versions is None:
# jitify version could be "<unknown>" and the angular brakets are not
# valid characters on Windows, so we need to strip them
_jitify_version = re.sub(r'<([^>]*)>', r'\1', get_build_version())
_jitify_cache_versions = (
f"{get_build_version()}_{cub.get_build_version()}_"
f"{_jitify_version}_{cub.get_build_version()}_"
f"{get_cuda_version()}_{build_num}_{cupy_cache_key.decode()}")

_init_cupy_headers()
Expand Down

0 comments on commit 666319b

Please sign in to comment.