From 779930ae524efbd5fd461bac9b9c37a120551178 Mon Sep 17 00:00:00 2001 From: Lixun Zhang Date: Fri, 12 Apr 2024 21:50:18 -0500 Subject: [PATCH] Rename env var --- include/triton/Tools/Sys/GetEnv.hpp | 2 +- lib/Target/HSACO/HSACOTranslation.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/include/triton/Tools/Sys/GetEnv.hpp b/include/triton/Tools/Sys/GetEnv.hpp index 7169ba4c027f..4650950954c2 100644 --- a/include/triton/Tools/Sys/GetEnv.hpp +++ b/include/triton/Tools/Sys/GetEnv.hpp @@ -32,7 +32,7 @@ namespace mlir::triton { const std::set 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 index a4192d69e93a..9d5de2514e51 100644 --- 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 args; args.push_back((char *)("triton"));