Add explicit conversion to nb::int

jorickert · jorickert · commit c3e69eea55ea · 2025-03-17T09:44:42.000-06:00
diff --git a/mlir/lib/Bindings/Python/IRAttributes.cpp b/mlir/lib/Bindings/Python/IRAttributes.cpp
@@ -1389,7 +1389,7 @@ class PyDenseIntElementsAttribute
     // TODO: consider caching the type properties in the constructor to avoid
     // querying them on each element access.
     if (mlirTypeIsAIndex(type)) {
-      return mlirDenseElementsAttrGetIndexValue(*this, pos);
+      return nb::int_(mlirDenseElementsAttrGetIndexValue(*this, pos));
     }
     unsigned width = mlirIntegerTypeGetWidth(type);
     bool isUnsigned = mlirIntegerTypeIsUnsigned(type);

Original file line number	Diff line number	Diff line change
`@@ -1389,7 +1389,7 @@ class PyDenseIntElementsAttribute`
`1389`	`1389`	`// TODO: consider caching the type properties in the constructor to avoid`
`1390`	`1390`	`// querying them on each element access.`
`1391`	`1391`	`if (mlirTypeIsAIndex(type)) {`
`1392`		`- return mlirDenseElementsAttrGetIndexValue(*this, pos);`
	`1392`	`+ return nb::int_(mlirDenseElementsAttrGetIndexValue(*this, pos));`
`1393`	`1393`	`}`
`1394`	`1394`	`unsigned width = mlirIntegerTypeGetWidth(type);`
`1395`	`1395`	`bool isUnsigned = mlirIntegerTypeIsUnsigned(type);`