Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Use CMake standard library to handle CUDA #17031

Merged
merged 1 commit into from
Dec 30, 2019
Merged

Use CMake standard library to handle CUDA #17031

merged 1 commit into from
Dec 30, 2019

Commits on Dec 27, 2019

  1. Switch to modern CMake CUDA handling 

    Introduce unified MXNET_CUDA_ARCH option to specify cuda architectures.
Previously cuda architecture setting was partially broken and different options
were applied to different parts of the build (CUDA_ARCH_NAME CUDA_ARCH_BIN
CUDA_ARCH_PTX and CUDA_ARCH_LIST).

Include FindCUDAToolkit from CMake 3.17, which replaces the deprecated FindCUDA
functionality for finding the cuda toolkit include directories and libraries.
    @leezu
    leezu committed Dec 27, 2019
    Configuration menu
    Copy the full SHA
    d949e0c View commit details
    Browse the repository at this point in the history