[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/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
+++ b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
@@ -1051,7 +1051,7 @@ struct LoadOpConversion
      if (other) {
        for (size_t ii = 0; ii < nWords; ++ii) {
          PTXInstr &mov = *ptxBuilder.create<>("mov");
-          mov.o("u", width);
+          mov.o("u" + std::to_string(width));

          size_t size = width / valueElemNbits;