Please use GitHub pull requests for new patches. Phabricator shutdown timeline
Changeset View
Changeset View
Standalone View
Standalone View
mlir/test/Dialect/ArmSME/roundtrip.mlir
- This file was added.
| // RUN: mlir-opt %s | |||||
| func.func @arm_sme_ops(%0 : vector<[2]xf64>, | |||||
| %1 : vector<[4]xf32>, | |||||
| %2 : vector<[4x2]xf16>, | |||||
| %3 : vector<[2x4]xsi16>) { | |||||
| %c = arith.constant 128 : index | |||||
| %pred.64 = vector.create_mask %c : vector<[2]xi1> | |||||
| arm_sme.zero za0d, za1d, za2d, za3d, za4d, za5d, za6d, za7d, za0s, za1s, za2s, za3s | |||||
| arm_sme.mopa za0d, %pred.64, %pred.64, %0, %0 : vector<[2]xi1>, vector<[2]xi1>, vector<[2]xf64>, vector<[2]xf64> | |||||
| %pred.32 = vector.create_mask %c : vector<[4]xi1> | |||||
| arm_sme.mopa za0s, %pred.32, %pred.32, %1, %1 : vector<[4]xi1>, vector<[4]xi1>, vector<[4]xf32>, vector<[4]xf32> | |||||
| %pred.16 = vector.create_mask %c, %c : vector<[4x2]xi1> | |||||
| arm_sme.mopa za1s, %pred.16, %pred.16, %2, %2 : vector<[4x2]xi1>, vector<[4x2]xi1>, vector<[4x2]xf16>, vector<[4x2]xf16> | |||||
| %pred.i16 = vector.create_mask %c, %c : vector<[2x4]xi1> | |||||
| arm_sme.mopa za1d, %pred.i16, %pred.i16, %3, %3 : vector<[2x4]xi1>, vector<[2x4]xi1>, vector<[2x4]xsi16>, vector<[2x4]xsi16> | |||||
| return | |||||
| } | |||||