Add a check if environment is not enable.

tenstorrent · Oct 4, 2024 · a5114d5 · a5114d5
1 parent d45dfed
commit a5114d5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -10,6 +10,9 @@
 
 cmake_minimum_required(VERSION 3.20)
 
+if (NOT DEFINED ENV{TTXLA_ENV_ACTIVATED})
+  message(FATAL_ERROR "tt-xla environment not activated. Please run 'source venv/activate'.")
+endif()
 
 project(TT_PJRT)
 cmake_policy(SET CMP0069 NEW)

diff --git a/venv/activate b/venv/activate
@@ -20,6 +20,7 @@ else
   pip install --upgrade pip
   pip install -r requirements.txt
 fi
+export TTXLA_ENV_ACTIVATED=1
 export TTMLIR_ENV_ACTIVATED=1
 export PATH=$TTMLIR_TOOLCHAIN_DIR/bin:$PATH
 if [ -n "$PROJECT_ROOT" ]; then