[Mlir-commits] [mlir] [NVGPU] Fix nvdsl examples (PR #156830)

[Giacomo Castiglioni]

================
@@ -35,9 +35,11 @@ def compile(self, module: ir.Module):
 
     def jit(self, module: ir.Module) -> execution_engine.ExecutionEngine:
         """Wraps the module in a JIT execution engine."""
-        return execution_engine.ExecutionEngine(
+        ee = execution_engine.ExecutionEngine(
             module, opt_level=self.opt_level, shared_libs=self.shared_libs
         )
+        ee.initialize()
----------------
castigli wrote:

As I tried to describe in the PR description, this is needed a side effect of the GPU module being loaded as a [global constructor.](https://github.com/castigli/llvm-project/blob/d1e4ae57c3a6c6f45eaae87db0b935a86e37047f/mlir/lib/Bindings/Python/ExecutionEngineModule.cpp#L132)


https://github.com/llvm/llvm-project/pull/156830