Merge pull request #1132 from ZenithalHourlyRate:fix-ntt-perf

PiperOrigin-RevId: 701960848
google · Dec 2, 2024 · 72d7242 · 72d7242
2 parents 7682bda + 1eee848
commit 72d7242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../Dialect/Polynomial/Conversions/heir_polynomial_to_llvm/runner/lower_ntt_perf_runner.mlir b/.../Dialect/Polynomial/Conversions/heir_polynomial_to_llvm/runner/lower_ntt_perf_runner.mlir
@@ -30,7 +30,7 @@ func.func @test_poly_ntt() {
 
   %2 = polynomial.to_tensor %1 : !poly_ty -> tensor<65536x!coeff_ty>
   %ext2 = mod_arith.extract %2 : tensor<65536x!coeff_ty> -> tensor<65536xi32>
-  %4 = bufferization.to_memref %ext2 : memref<65536xi32>
+  %4 = bufferization.to_memref %ext2 : tensor<65536xi32> to memref<65536xi32>
   %U = memref.cast %4 : memref<65536xi32> to memref<*xi32>
   func.call @printMemrefI32(%U) : (memref<*xi32>) -> ()
   return