From d50e12a37b97f4efd59878140a100f5fd8bc4305 Mon Sep 17 00:00:00 2001
From: Evgeny Mankov <evgeny.mankov@gmail.com>
Date: Mon, 18 Nov 2024 16:56:15 +0100
Subject: [PATCH] [HIPIFY][Tensor][tests][fix] Do not run `Tensor` tests if
 `CUDA_TENSOR_ROOT_DIR` is not specified

---
 CMakeLists.txt | 2 +-
 tests/lit.cfg  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index be1c01c6..44b0f17e 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -335,7 +335,7 @@ if(HIPIFY_CLANG_TESTS OR HIPIFY_CLANG_TESTS_ONLY)
   endif()
 
   if(CUDA_TENSOR_ROOT_DIR STREQUAL "")
-      set(CUDA_TENSOR_ROOT_DIR OFF)
+    set(CUDA_TENSOR_ROOT_DIR OFF)
   endif()
 
   if(CUDA_CUB_ROOT_DIR STREQUAL "")
diff --git a/tests/lit.cfg b/tests/lit.cfg
index 678cc760..7bffc581 100644
--- a/tests/lit.cfg
+++ b/tests/lit.cfg
@@ -40,6 +40,10 @@ if not config.cuda_cub_root or config.cuda_cub_root == "OFF":
     config.excludes.append('cub_03.cu')
     print("WARN:   CUB tests are excluded because CUDA_CUB_ROOT_DIR is not specified")
     warns = True
+if not config.cuda_tensor_root or config.cuda_tensor_root == "OFF":
+    config.excludes.append('cutensor2hiptensor.cu')
+    print("WARN:   cuTensor tests are excluded because CUDA_TENSOR_ROOT_DIR is not specified")
+    warns = True
 if warns:
     print(delimiter)