[Mlir-commits] [mlir] [mlir][tosa] Allow zero-points to be unranked (PR #143770)
Luke Hutton
llvmlistbot at llvm.org
Mon Jun 16 07:02:35 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>
----------------
lhutton1 wrote:
The tosa-infer-shapes pass doesn't rewrite the types of input parameters, so unfortunately we cannot avoid the cast completely. We could however avoid the consts - I've updated the test case accordingly.
https://github.com/llvm/llvm-project/pull/143770
More information about the Mlir-commits
mailing list