Skip to content

Commit 8370503

Browse files
committed
fix python bindings
1 parent 9814aec commit 8370503

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/lib/Bindings/Python/DialectGPU.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ PYBIND11_MODULE(_mlirDialectsGPU, m) {
5353
py::buffer_info info(py::buffer(object).request());
5454
MlirStringRef objectStrRef =
5555
mlirStringRefCreate(static_cast<char *>(info.ptr), info.size);
56-
return cls(mlirGPUObjectAttrGet(
56+
return cls(mlirGPUObjectAttrGetWithKernels(
5757
mlirAttributeGetContext(target), target, format, objectStrRef,
5858
mlirObjectProps.has_value() ? *mlirObjectProps
5959
: MlirAttribute{nullptr},

0 commit comments

Comments
 (0)