CI

Signed-off-by: Serge Panev <[email protected]>

CMakeLists.txt

@@ -332,6 +332,7 @@ include_directories("3rdparty/tvm/include")
 include_directories("3rdparty/dmlc-core/include")
 include_directories("3rdparty/dlpack/include")
 
+message(STATUS "CUDAToolkit_VERSION_MAJOR ${CUDAToolkit_VERSION_MAJOR}")
 if(CUDAToolkit_VERSION_MAJOR LESS "11")
   include_directories("3rdparty/nvidia_cub")
 endif()

cmake/upstream/FindCUDAToolkit.cmake

@@ -663,6 +663,8 @@ if(NOT CUDAToolkit_BIN_DIR AND CUDAToolkit_NVCC_EXECUTABLE)
   unset(cuda_dir)
 endif()
 
+message(STATUS "CUDAToolkit_NVCC_EXECUTABLE ${CUDAToolkit_NVCC_EXECUTABLE}")
+message(STATUS "CMAKE_CUDA_COMPILER_VERSION ${CMAKE_CUDA_COMPILER_VERSION}")
 if(CUDAToolkit_NVCC_EXECUTABLE AND
    CUDAToolkit_NVCC_EXECUTABLE STREQUAL CMAKE_CUDA_COMPILER)
   # Need to set these based off the already computed CMAKE_CUDA_COMPILER_VERSION value
@@ -676,6 +678,7 @@ if(CUDAToolkit_NVCC_EXECUTABLE AND
 else()
   # Compute the version by invoking nvcc
   execute_process (COMMAND ${CUDAToolkit_NVCC_EXECUTABLE} "--version" OUTPUT_VARIABLE NVCC_OUT)
+  message(STATUS "NVCC_OUT ${NVCC_OUT}")
   if(NVCC_OUT MATCHES [=[ V([0-9]+)\.([0-9]+)\.([0-9]+)]=])
     set(CUDAToolkit_VERSION_MAJOR "${CMAKE_MATCH_1}")
     set(CUDAToolkit_VERSION_MINOR "${CMAKE_MATCH_2}")

python/mxnet/contrib/__init__.py

@@ -32,4 +32,3 @@
 from . import io
 from . import quantization
 from . import quantization as quant
-from . import tensorrt