Add call arg test + fix crash

Author: nhat-nguyen

lib/Conversion/TritonPtrToMemref/TritonPtrToMemrefPass.cpp

@@ -6,6 +6,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "mlir/Dialect/Arith/IR/Arith.h"
+#include "mlir/Dialect/Func/Transforms/FuncConversions.h"
 #include "mlir/Dialect/MemRef/IR/MemRef.h"
 #include "mlir/Dialect/SCF/IR/SCF.h"
 #include "mlir/IR/Builders.h"
@@ -102,8 +103,7 @@ class TritonPtrToMemrefPass
         patterns, typeConverter);
     populateFunctionOpInterfaceTypeConversionPattern<triton::FuncOp>(
         patterns, typeConverter);
-    populateFunctionOpInterfaceTypeConversionPattern<func::CallOp>(
-        patterns, typeConverter);
+    populateCallOpTypeConversionPattern(patterns, typeConverter);
 
     if (failed(applyPartialConversion(moduleOp, target, std::move(patterns)))) {
       signalPassFailure();

test/Conversion/TritonPtrToMemref/call.mlir

@@ -0,0 +1,23 @@
+// RUN: triton-shared-opt --triton-arith-to-linalg --triton-ptr-to-memref %s | FileCheck %s
+
+module {
+  tt.func @_sum_combine__fp32(%arg0: !tt.ptr<f32>) -> f32{
+    %0 = arith.constant 42.0 : f32
+    tt.return %0 : f32
+  }
+  tt.func @test(%arg0: !tt.ptr<f32>) -> f32{
+    %0 = tt.call @_sum_combine__fp32(%arg0) : (!tt.ptr<f32>) -> f32
+    tt.return %0 : f32
+  }
+}
+
+// CHECK: module {
+// CHECK:   func.func @_sum_combine__fp32(%arg0: memref<*xf32>, %arg1: i32, %arg2: i32, %arg3: i32, %arg4: i32, %arg5: i32, %arg6: i32) -> f32 {
+// CHECK:     %cst = arith.constant 4.200000e+01 : f32
+// CHECK:     return %cst : f32
+// CHECK:   }
+// CHECK:   func.func @test(%arg0: memref<*xf32>, %arg1: i32, %arg2: i32, %arg3: i32, %arg4: i32, %arg5: i32, %arg6: i32) -> f32 {
+// CHECK:     %0 = call @_sum_combine__fp32(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6) : (memref<*xf32>, i32, i32, i32, i32, i32, i32) -> f32
+// CHECK:     return %0 : f32
+// CHECK:   }
+// CHECK: }