microsoft/onnxruntime-extensions

Public

revert-703-bsrc

Find a branch or tag

HTTPS

cmake/ext_cuda.cmake

24lines · modecode

unknown

1	`# Copyright (c) Microsoft Corporation. All rights reserved.`
2	`# Licensed under the MIT License.`
3
4	`find_package(CUDAToolkit)`
5	`enable_language(CUDA)`
6
7	`set(CMAKE_CUDA_RUNTIME_LIBRARY Shared)`
8	`set(CMAKE_CUDA_STANDARD 17)`
9
10	`set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-relaxed-constexpr")`
11	`if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 11)`
12	`set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --Werror default-stream-launch")`
13	`endif()`
14
15	`if(NOT WIN32)`
16	`list(APPEND CUDA_NVCC_FLAGS --compiler-options -fPIC)`
17	`endif()`
18
19	`# Options passed to cudafe`
20	`set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=bad_friend_decl\"")`
21	`set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=unsigned_compare_with_zero\"")`
22	`set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=expr_has_no_effect\"")`
23
24	`add_compile_definitions(USE_CUDA)`
25