[Mlir-commits] [mlir] [mlir][tosa] Allow zero-points to be unranked (PR #143770)

Peng Sun llvmlistbot at llvm.org
Mon Jun 16 05:45:57 PDT 2025


================
@@ -333,6 +333,19 @@ func.func @test_dynamic_mixed_matmul(%arg0 : tensor<?x3x?xi32>, %arg1 : tensor<?
 
 // -----
 
+// CHECK-LABEL: @test_unranked_zero_points_matmul
+func.func @test_unranked_zero_points_matmul(%arg0: tensor<1x2x3xf32>, %arg1: tensor<1x3x4xf32>) -> tensor<1x2x4xf32> {
+    %a_zp = "tosa.const"() <{values = dense<0> : tensor<1xi8>}> : () -> tensor<1xi8>
+    %b_zp = "tosa.const"() <{values = dense<0> : tensor<1xi8>}> : () -> tensor<1xi8>
----------------
psunn wrote:

NIT:
Is it necessary to retain the `tosa.const` followed by `tosa.cast` in this test case?

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


More information about the Mlir-commits mailing list