extract function to c++

Author: grypp

mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td

@@ -155,11 +155,7 @@ class NVVM_SpecialRangeableRegisterOp<string mnemonic> :
     void $cppClass::inferResultRanges(
         ArrayRef<::mlir::ConstantIntRanges> argRanges,
         SetIntRangeFn setResultRanges) {
-        if (auto rangeAttr = getOperation()->getAttrOfType<LLVM::ConstantRangeAttr>("range")) {
-          setResultRanges(getResult(), 
-                          {rangeAttr.getLower(), rangeAttr.getUpper(),
-                          rangeAttr.getLower(), rangeAttr.getUpper()});
-        }
+        nvvmInferResultRanges(getOperation(), getResult(), argRanges, setResultRanges);
     }
   }];
 

mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp

@@ -1158,6 +1158,17 @@ llvm::Intrinsic::ID CpAsyncBulkTensorReduceOp::getIntrinsicID(
   llvm_unreachable("Invalid Reduction Op for CpAsyncBulkTensorReduceOp");
 }
 
+/// Infer the result ranges for the NVVM SpecialRangeableRegisterOp that migth
+/// have ConstantRangeAttr.
+static void nvvmInferResultRanges(Operation *op, Value result,
+                                  ArrayRef<::mlir::ConstantIntRanges> argRanges,
+                                  SetIntRangeFn setResultRanges) {
+  if (auto rangeAttr = op->getAttrOfType<LLVM::ConstantRangeAttr>("range")) {
+    setResultRanges(result, {rangeAttr.getLower(), rangeAttr.getUpper(),
+                             rangeAttr.getLower(), rangeAttr.getUpper()});
+  }
+}
+
 //===----------------------------------------------------------------------===//
 // NVVMDialect initialization, type parsing, and registration.
 //===----------------------------------------------------------------------===//