Skip to content

Commit

Permalink
Rename env var
Browse files Browse the repository at this point in the history
  • Loading branch information
zhanglx13 committed Apr 13, 2024
1 parent 0dbd13c commit 779930a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion include/triton/Tools/Sys/GetEnv.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -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 {

Expand Down
2 changes: 1 addition & 1 deletion lib/Target/HSACO/HSACOTranslation.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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"));
Expand Down

0 comments on commit 779930a

Please sign in to comment.