Skip to content

Commit 6b7ffb8

Browse files
add some debug info
1 parent 1f732e4 commit 6b7ffb8

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

python/cudf/cudf/utils/_numba.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import glob
55
import os
66
import sys
7+
import warnings
78
from functools import lru_cache
89

910
import numba
@@ -119,6 +120,7 @@ def _setup_numba():
119120
from cudf.utils._ptxcompiler import NO_DRIVER, safe_get_versions
120121

121122
versions = safe_get_versions()
123+
warnings.warn(f"versions: {versions}\n")
122124
if versions != NO_DRIVER:
123125
driver_version, runtime_version = versions
124126
shim_ptx_cuda_version = _get_cuda_build_version()
@@ -127,6 +129,11 @@ def _setup_numba():
127129
# This could be the shipped shim PTX file (determined by the CUDA
128130
# version used at build time) or the PTX emitted by the version of NVVM
129131
# on the user system (determined by the user's CUDA runtime version)
132+
warnings.warn(
133+
f"driver_version: {driver_version},"
134+
f"runtime_version: {runtime_version},",
135+
f"shim_ptx_cuda_version: {shim_ptx_cuda_version}\n",
136+
)
130137
if (driver_version < shim_ptx_cuda_version) or (
131138
driver_version < runtime_version
132139
):

0 commit comments

Comments
 (0)