Fix the LLVM build

triton-lang · Jan 7, 2025 · 1ccc3f6 · 1ccc3f6
1 parent d13a040
commit 1ccc3f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/third_party/nvidia/lib/NVGPUToLLVM/NVGPUToLLVMPass.cpp b/third_party/nvidia/lib/NVGPUToLLVM/NVGPUToLLVMPass.cpp
@@ -9,6 +9,7 @@
 #include "mlir/Transforms/GreedyPatternRewriteDriver.h"
 
 #include "nvidia/lib/TritonNVIDIAGPUToLLVM/Utility.h"
+#include "llvm/Support/ErrorHandling.h"
 
 using namespace mlir;
 using namespace mlir::triton;
@@ -271,7 +272,7 @@ class MatrixOpPattern : public OpRewritePattern<MatrixOpType> {
     case 4:
       return (ptxAsmBase + ".x4" + suffix).str();
     default:
-      assert(false && "Invalid vector size");
+      llvm_unreachable("Invalid vector size");
     }
   }