xref: /llvm-project/mlir/test/Dialect/ArmSME/enable-arm-za.mlir (revision 1a498103ee5c4d101e70dc49db11938d8b87b518)
1// RUN: mlir-opt %s -enable-arm-streaming=za-mode=new-za | FileCheck %s -check-prefix=ENABLE-ZA
2// RUN: mlir-opt %s -enable-arm-streaming | FileCheck %s -check-prefix=DISABLE-ZA
3// RUN: mlir-opt %s -enable-arm-streaming=za-mode=in-za | FileCheck %s -check-prefix=IN-ZA
4// RUN: mlir-opt %s -enable-arm-streaming=za-mode=out-za | FileCheck %s -check-prefix=OUT-ZA
5// RUN: mlir-opt %s -enable-arm-streaming=za-mode=inout-za | FileCheck %s -check-prefix=INOUT-ZA
6// RUN: mlir-opt %s -enable-arm-streaming=za-mode=preserves-za | FileCheck %s -check-prefix=PRESERVES-ZA
7
8// CHECK-LABEL: @declaration
9func.func private @declaration()
10
11// ENABLE-ZA-LABEL: @arm_new_za
12// ENABLE-ZA-SAME: attributes {arm_new_za, arm_streaming}
13// IN-ZA-LABEL: @arm_new_za
14// IN-ZA-SAME: attributes {arm_in_za, arm_streaming}
15// OUT-ZA-LABEL: @arm_new_za
16// OUT-ZA-SAME: attributes {arm_out_za, arm_streaming}
17// INOUT-ZA-LABEL: @arm_new_za
18// INOUT-ZA-SAME: attributes {arm_inout_za, arm_streaming}
19// PRESERVES-ZA-LABEL: @arm_new_za
20// PRESERVES-ZA-SAME: attributes {arm_preserves_za, arm_streaming}
21// DISABLE-ZA-LABEL: @arm_new_za
22// DISABLE-ZA-NOT: arm_new_za
23// DISABLE-ZA-SAME: attributes {arm_streaming}
24func.func @arm_new_za() { return }
25