File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed
third_party/nvidia/lib/TritonNVIDIAGPUToLLVM Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -172,17 +172,20 @@ std::string PTXInstrExecution::dump() const {
172172 std::string osStr;
173173 llvm::raw_string_ostream os (osStr);
174174
175- std::string instrRepr = strJoin (instr->instrParts , " ." );
176- if (onlyAttachMLIRArgs)
177- return instrRepr;
178-
179175 if (pred) {
180176 if (!pred->repr )
181177 os << " @" << pred->dump () << " " ;
182178 else
183179 os << pred->repr (pred->idx ) << " " ;
184180 }
185181
182+ std::string instrRepr = strJoin (instr->instrParts , " ." );
183+ if (onlyAttachMLIRArgs) {
184+ os << instrRepr;
185+ os.flush ();
186+ return osStr;
187+ }
188+
186189 llvm::SmallVector<std::string, 4 > argReprs;
187190 for (auto *arg : argsInOrder) {
188191 argReprs.push_back (arg->dump ());
You can’t perform that action at this time.
0 commit comments