# RUN: %PYTHON %s | FileCheck %s

from mlir.ir import *
import mlir.dialects.std as std
import mlir.dialects.memref as memref

def run(f):
  print("\nTEST:", f.__name__)
  f()

# CHECK-LABEL: TEST: testSubViewAccessors
def testSubViewAccessors():
  ctx = Context()
  module = Module.parse(r"""
    func @f1(%arg0: memref<?x?xf32>) {
      %0 = constant 0 : index
      %1 = constant 1 : index
      %2 = constant 2 : index
      %3 = constant 3 : index
      %4 = constant 4 : index
      %5 = constant 5 : index
      memref.subview %arg0[%0, %1][%2, %3][%4, %5] : memref<?x?xf32> to memref<?x?xf32, offset: ?, strides: [?, ?]>
      return
    }
  """, ctx)
  func_body = module.body.operations[0].regions[0].blocks[0]
  subview = func_body.operations[6]

  assert subview.source == subview.operands[0]
  assert len(subview.offsets) == 2
  assert len(subview.sizes) == 2
  assert len(subview.strides) == 2
  assert subview.result == subview.results[0]

  # CHECK: SubViewOp
  print(type(subview).__name__)

  # CHECK: constant 0
  print(subview.offsets[0])
  # CHECK: constant 1
  print(subview.offsets[1])
  # CHECK: constant 2
  print(subview.sizes[0])
  # CHECK: constant 3
  print(subview.sizes[1])
  # CHECK: constant 4
  print(subview.strides[0])
  # CHECK: constant 5
  print(subview.strides[1])


run(testSubViewAccessors)

