[Triton-MLIR] Minor fixes to enable fused-softmax and layer-norm tutorials (#835)

2022-11-09 10:18:56 +08:00
parent 2da71b2aaa
commit e517b58d59
4 changed files with 5 additions and 5 deletions
--- a/include/triton/Conversion/TritonGPUToLLVM/PtxAsmFormat.h
+++ b/include/triton/Conversion/TritonGPUToLLVM/PtxAsmFormat.h
@@ -320,7 +320,7 @@ struct PTXInstrExecution {
  // Prefix a !predicate to the instruction.
  PTXInstrExecution &predicateNot(mlir::Value value, StringRef constraint) {
    pred = instr->builder->newOperand(value, constraint);
-    pred->repr = [](int idx) { return "@!%" + std::to_string(idx); };
+    pred->repr = [](int idx) { return "@!$" + std::to_string(idx); };
    return *this;
  }