gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] Replace the legacy print call in triton.cc with the SlotTracker-based one. (#396)

Browse Source 
The legacy print call will assign names (e.g., %10) to values, which can be undesirable in some cases.
This commit is contained in:
daadaada
2021-12-19 10:03:22 +08:00
committed by GitHub
parent fa62b4a8f6
commit 4a8953efa3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/triton.cc
View File

@@ -440,7 +440,7 @@ void init_triton_codegen(py::module &&m) {
// record asm as we generate // record asm as we generate
asm_map_t asm_map; asm_map_t asm_map;
std::ostringstream ttir; std::ostringstream ttir;
ir::print(ir, ttir); ir.print(ttir);
asm_map["ttir"] = py::cast(ttir.str()); asm_map["ttir"] = py::cast(ttir.str());
llvm::LLVMContext ctx; llvm::LLVMContext ctx;
if(backend == CUDA) if(backend == CUDA)