Skip to content

Commit 96d167e

Browse files
level_type for pybind
1 parent b7a0aba commit 96d167e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mlir/lib/Bindings/Python/DialectSparseTensor.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ static void populateDialectSparseTensorSubmodule(const py::module &m) {
4646
mlirAttributeIsASparseTensorEncodingAttr)
4747
.def_classmethod(
4848
"get",
49-
[](py::object cls, std::vector<MlirSparseTensorLevelType> lvlTypes,
49+
[](py::object cls, std::vector<level_type> lvlTypes,
5050
std::optional<MlirAffineMap> dimToLvl,
5151
std::optional<MlirAffineMap> lvlToDim, int posWidth, int crdWidth,
5252
MlirContext context) {
@@ -64,7 +64,7 @@ static void populateDialectSparseTensorSubmodule(const py::module &m) {
6464
"lvl_types",
6565
[](MlirAttribute self) {
6666
const int lvlRank = mlirSparseTensorEncodingGetLvlRank(self);
67-
std::vector<MlirSparseTensorLevelType> ret;
67+
std::vector<level_type> ret;
6868
ret.reserve(lvlRank);
6969
for (int l = 0; l < lvlRank; ++l)
7070
ret.push_back(mlirSparseTensorEncodingAttrGetLvlType(self, l));

0 commit comments

Comments
 (0)