Rename env var

ROCm · Apr 13, 2024 · 779930a · 779930a
1 parent 0dbd13c
commit 779930a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/include/triton/Tools/Sys/GetEnv.hpp b/include/triton/Tools/Sys/GetEnv.hpp
@@ -32,7 +32,7 @@ namespace mlir::triton {
 const std::set<std::string> ENV_VARS = {
     "DISABLE_MMA_V3",     "TRITON_DISABLE_LINE_INFO", "DISABLE_FAST_REDUCTION",
     "ENABLE_TMA",         "MLIR_ENABLE_DUMP",         "LLVM_IR_ENABLE_DUMP",
-    "AMDGCN_ENABLE_DUMP", "TRUNCATE_F32_TO_BF16",     "LLVM_FLAG"};
+    "AMDGCN_ENABLE_DUMP", "TRUNCATE_F32_TO_BF16",     "TRITON_LLVM_FLAG"};
 
 namespace tools {
 

diff --git a/lib/Target/HSACO/HSACOTranslation.cpp b/lib/Target/HSACO/HSACOTranslation.cpp
@@ -135,7 +135,7 @@ std::string generate_amdgcn_assembly(llvm::Module *module,
   if (machine == nullptr)
     return "";
 
-  std::string llvm_flag = mlir::triton::tools::getenv("LLVM_FLAG");
+  std::string llvm_flag = mlir::triton::tools::getenv("TRITON_LLVM_FLAG");
   if (!llvm_flag.empty()) {
     std::vector<char *> args;
     args.push_back((char *)("triton"));