Skip to content

Commit

Permalink
add a warning as a stop-gap solution...
Browse files Browse the repository at this point in the history
  • Loading branch information
leofang committed Mar 11, 2024
1 parent 8fb6d04 commit 5528202
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions cupy/cuda/jitify.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,11 @@ import json
import os
import re
import tempfile
import warnings

from cupy._environment import get_cuda_path
from cupy.cuda import cub
from cupy import _util


###############################################################################
Expand Down Expand Up @@ -186,6 +188,12 @@ cdef inline void _init_cupy_headers_from_scratch() except*:
# Same for tuple
cupy_headers[b"tuple"] = b"#include <cupy/cuda_workaround.h>\n"

# Ensure users know this is normal and not hanging...
warnings.warn(
"Jitify is performing a one-time only warm-up to populate the "
"persistent cache, this may take a few seconds and will be improved "
"in a future release...", _util.PerformanceWarning)

# Compile a dummy kernel to further populate the cache (with bundled
# headers)
# need to defer import to avoid circular dependency
Expand Down

0 comments on commit 5528202

Please sign in to comment.