[SYCL-MLIR][cgeist] Do not use SingleBlock::push_back on gpu.module

This leads to pushing operations after the terminator.

Signed-off-by: Victor Perez <[email protected]>

Author: victor-eds

polygeist/tools/cgeist/Lib/clang-mlir.cc

@@ -2148,11 +2148,13 @@ MLIRASTConsumer::createMLIRFunction(const FunctionToEmit &FTE,
     Module->push_back(Function);
     Functions[MangledName] = cast<func::FuncOp>(Function);
     break;
-  case InsertionContext::SYCLDevice:
-    mlirclang::getDeviceModule(*Module).push_back(Function);
+  case InsertionContext::SYCLDevice: {
+    gpu::GPUModuleOp deviceModule = mlirclang::getDeviceModule(*Module);
+    deviceModule.insert(deviceModule.getBody()->getTerminator(), Function);
     DeviceFunctions[MangledName] = Function;
     break;
   }
+  }
 
   LLVM_DEBUG(llvm::dbgs() << "Created MLIR function: " << Function << "\n");