// RUN: tutorial-opt %s --loop-invariant-code-motion > %t
// RUN: FileCheck %s < %t

module {
  // CHECK-LABEL: func.func @test_loop_invariant_code_motion
  func.func @test_loop_invariant_code_motion() -> !poly.poly<10> {
    %0 = arith.constant dense<[1, 2, 3]> : tensor<3xi32>
    %p0 = poly.from_tensor %0 : tensor<3xi32> -> !poly.poly<10>

    %1 = arith.constant dense<[9, 8, 16]> : tensor<3xi32>
    %p1 = poly.from_tensor %1 : tensor<3xi32> -> !poly.poly<10>
    // CHECK: poly.mul

    // CHECK: affine.for
    %ret_val = affine.for %i = 0 to 100 iter_args(%sum_iter = %p0) -> !poly.poly<10> {
      // The poly.mul should be hoisted out of the loop.
      // CHECK-NOT: poly.mul
      %2 = poly.mul %p0, %p1 : !poly.poly<10>
      %sum_next = poly.add %sum_iter, %2 : !poly.poly<10>
      affine.yield %sum_next : !poly.poly<10>
    }

    return %ret_val : !poly.poly<10>
  }
}