Update onnxruntime_unittests.cmake (microsoft#10215)

garymm · Jan 8, 2022 · 5ebb857 · 5ebb857
1 parent bacae96
commit 5ebb857
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmake/onnxruntime_unittests.cmake b/cmake/onnxruntime_unittests.cmake
@@ -1170,6 +1170,9 @@ if (onnxruntime_USE_CUDA)
   onnxruntime_add_shared_library_module(custom_op_library ${ONNXRUNTIME_SHARED_LIB_TEST_SRC_DIR}/cuda_ops.cu
                                                           ${TEST_SRC_DIR}/testdata/custom_op_library/custom_op_library.cc)
   target_include_directories(custom_op_library PRIVATE ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
+  if (HAS_QSPECTRE)
+    target_compile_options(custom_op_library PRIVATE "$<$<COMPILE_LANGUAGE:CUDA>:SHELL:-Xcompiler /Qspectre>")
+  endif()
 else()
   onnxruntime_add_shared_library_module(custom_op_library ${TEST_SRC_DIR}/testdata/custom_op_library/custom_op_library.cc)
 endif()