legalize name for kern lowering

EnzymeAD · Jan 18, 2025 · 07a8928 · 07a8928
1 parent 03de92c
commit 07a8928
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/LowerKernel.cpp b/src/enzyme_ad/jax/Passes/LowerKernel.cpp
@@ -353,7 +353,9 @@ CallInfo CompileKernel(SymbolTableCollection &symbolTable, mlir::Location loc,
   auto gpumod = builder.create<gpu::GPUModuleOp>(loc, "gpumodname");
   builder.setInsertionPointToStart(&gpumod.getBodyRegion().front());
 
-  auto gpufunc = builder.create<gpu::GPUFuncOp>(loc, op.getName(), gpuTy);
+  std::string legalName = op.getName().str();
+  std::replace(legalName.begin(), legalName.end(), '#', '_');
+  auto gpufunc = builder.create<gpu::GPUFuncOp>(loc, legalName, gpuTy);
   {
     auto entry = &gpufunc.getBody().front();
     builder.setInsertionPointToEnd(entry);