Fix compilation errors in cnm-to-gpu and cinm-vulkan-runner.

tud-ccc · Sep 3, 2024 · 5b41ac8 · 5b41ac8
1 parent 3e39d1a
commit 5b41ac8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/cinnamon/lib/Conversion/CnmToGPU/CnmToGPU.cpp b/cinnamon/lib/Conversion/CnmToGPU/CnmToGPU.cpp
@@ -314,19 +314,19 @@ void populateCnmToGPUFinalTypeConversions(TypeConverter &typeConverter) {
       });
 }
 
-void populateCnmToGPUConversionPatterns(TypeConverter &typeConverter,
-                                        RewritePatternSet &patterns) {
+void populateCnmToGPUConversionPatterns(RewritePatternSet &patterns,
+                                        MLIRContext *context) {
   patterns
       .add<cnmtogpu::ConvertCnmWorkgroupToGPU, cnmtogpu::ConvertCnmAllocToGPU,
            ConvertCnmSetZeroToAffine, cnmtogpu::ConvertCnmScatterToGPU,
            cnmtogpu::ConvertCnmGatherToGPU, cnmtogpu::ConvertCnmLaunchToGPU,
-           cnmtogpu::ConvertCnmTerminatorToGPU>(&typeConverter.getContext());
+           cnmtogpu::ConvertCnmTerminatorToGPU>(context);
 }
 
 struct ConvertCnmToGPUPass
     : public ::impl::ConvertCnmToGPUPassBase<ConvertCnmToGPUPass> {
   void runOnOperation() final {
-    TypeConverter converter(&getContext());
+    TypeConverter converter{};
     populateCnmToGPUFinalTypeConversions(converter);
     const auto addUnrealizedCast = [](OpBuilder &builder, Type type,
                                       ValueRange inputs, Location loc) {
@@ -337,7 +337,7 @@ struct ConvertCnmToGPUPass
     converter.addTargetMaterialization(addUnrealizedCast);
 
     RewritePatternSet patterns(&getContext());
-    populateCnmToGPUConversionPatterns(converter, patterns);
+    populateCnmToGPUConversionPatterns(patterns, &getContext());
     populateReconcileUnrealizedCastsPatterns(patterns);
 
     ConversionTarget target(getContext());

diff --git a/cinnamon/tools/cinm-vulkan-runner/cinm-vulkan-runner.cpp b/cinnamon/tools/cinm-vulkan-runner/cinm-vulkan-runner.cpp
@@ -59,15 +59,15 @@ static LogicalResult runMLIRPasses(Operation *op) {
   passManager.addPass(createLowerAffinePass()); // affine.apply -> arith ops
 
   passManager.addPass(createCnmSPIRVAttachTargetAttributePass(
-      CnmSPIRVAttachTargetAttributePassOptions{
+      cnm::CnmSPIRVAttachTargetAttributePassOptions{
           .spirvCapabilities = {"Shader"},
           .spirvExtensions = {"SPV_KHR_storage_buffer_storage_class"},
       }));
 
   OpPassManager &gpuModulePM = passManager.nest<gpu::GPUModuleOp>();
   gpuModulePM.addPass(createConvertMemRefToSPIRVPass());
   gpuModulePM.addPass(createConvertControlFlowToSPIRVPass());
-  gpuModulePM.addPass(createCnmSPIRVAttachKernelEntryPointAttributePass());
+  gpuModulePM.addPass(cnm::createCnmSPIRVAttachKernelEntryPointAttributePass());
 
   passManager.addPass(createConvertGPUToSPIRVPass(/*mapMemorySpace=*/true));