be consistent, either #if #elif or #if defined #elif defined

ROCm · Oct 9, 2023 · 6686321 · 6686321
1 parent 1251c46
commit 6686321
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/test/contrib_ops/greedy_search_test.cc b/onnxruntime/test/contrib_ops/greedy_search_test.cc
@@ -58,7 +58,7 @@ TEST(GreedySearchTest, GptGreedySearchFp16_VocabPadded) {
     Ort::SessionOptions session_options;
 #ifdef USE_CUDA
     Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CUDA(session_options, 0));
-#elif USE_ROCM
+#elif defined(USE_ROCM)
     Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_ROCM(session_options, 0));
 #endif
 
@@ -130,7 +130,7 @@ TEST(GreedySearchTest, GptGreedySearchFp32) {
     Ort::SessionOptions session_options;
 #ifdef USE_CUDA
     Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CUDA(session_options, 0));
-#elif USE_ROCM
+#elif defined(USE_ROCM)
     Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_ROCM(session_options, 0));
 #endif