File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44import glob
55import os
66import sys
7+ import warnings
78from functools import lru_cache
89
910import 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 ):
You can’t perform that action at this time.
0 commit comments