[FRONTEND][BACKEND] Fix bool and int8 load when the other operand is given (#968)

2022-12-08 11:52:18 -08:00
parent 71c35bcf9c
commit 83f3b9165b
3 changed files with 43 additions and 6 deletions
--- a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
+++ b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
@@ -973,8 +973,10 @@ struct LoadOpConversion

      if (other) {
        for (size_t ii = 0; ii < nWords; ++ii) {
+          // PTX doesn't support mov.u8, so we need to use mov.u16
+          auto movWidth = width < 16 ? 16 : width;
          PTXInstr &mov =
-              ptxBuilder.create<>("mov")->o("u" + std::to_string(width));
+              ptxBuilder.create<>("mov")->o("u" + std::to_string(movWidth));

          size_t size = width / valueElemNbits;