[Mlir-commits] [mlir] bcd08f4 - [mlir][llvmir] Add SameOperandsAndResultType trait to LLVM_CountZerosIntrOp (#131133)

[llvmlistbot at llvm.org]

Author: Letu Ren
Date: 2025-03-23T12:15:11+01:00
New Revision: bcd08f4d2cf660026a965e11a6a4c9845bf44d41

URL: https://github.com/llvm/llvm-project/commit/bcd08f4d2cf660026a965e11a6a4c9845bf44d41
DIFF: https://github.com/llvm/llvm-project/commit/bcd08f4d2cf660026a965e11a6a4c9845bf44d41.diff

LOG: [mlir][llvmir] Add SameOperandsAndResultType trait to LLVM_CountZerosIntrOp (#131133)

According to https://llvm.org/docs/LangRef.html#llvm-ctlz-intrinsic and
https://llvm.org/docs/LangRef.html#llvm-cttz-intrinsic, The return type
of `llvm.intr.ctlz` and `llvm.intr.cttz` must match the first argument
type.

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td b/mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td
index e80745781f566..37193f384174e 100644
--- a/mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td
+++ b/mlir/include/mlir/Dialect/LLVMIR/LLVMIntrinsicOps.td
@@ -86,7 +86,7 @@ class LLVM_TernarySameArgsIntrOpF<string func, list<Trait> traits = []> :
 
 class LLVM_CountZerosIntrOp<string func, list<Trait> traits = []> :
     LLVM_OneResultIntrOp<func, [], [0],
-           !listconcat([Pure], traits),
+           !listconcat([Pure, SameOperandsAndResultType], traits),
             /*requiresFastmath=*/0,
             /*immArgPositions=*/[1], /*immArgAttrNames=*/["is_zero_poison"]> {
   let arguments = (ins LLVM_ScalarOrVectorOf<AnySignlessInteger>:$in,