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

ROCm · Oct 9, 2023 · 1251c46 · 1251c46
1 parent 5bc0f95
commit 1251c46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/test/contrib_ops/fft_op_test.cc b/onnxruntime/test/contrib_ops/fft_op_test.cc
@@ -11,7 +11,7 @@ TEST(ContribOpTest, Rfft) {
   std::vector<std::unique_ptr<IExecutionProvider>> execution_providers;
 #ifdef USE_CUDA
   execution_providers.push_back(DefaultCudaExecutionProvider());
-#elif USE_ROCM
+#elif defined(USE_ROCM)
   execution_providers.push_back(DefaultRocmExecutionProvider());
 #else
   return;
@@ -31,7 +31,7 @@ TEST(ContribOpTest, Irfft) {
   std::vector<std::unique_ptr<IExecutionProvider>> execution_providers;
 #ifdef USE_CUDA
   execution_providers.push_back(DefaultCudaExecutionProvider());
-#elif USE_ROCM
+#elif defined(USE_ROCM)
   execution_providers.push_back(DefaultRocmExecutionProvider());
 #else
   return;