================ @@ -4,3 +4,4 @@ from ._nvgpu_ops_gen import * from ._nvgpu_enum_gen import * +from .._mlir_libs._mlirDialectsNvgpu import * ---------------- ftynse wrote: Nit: trailing `\n` plz. https://github.com/llvm/llvm-project/pull/88855