124 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 52; RUN: opt -S -passes=instcombine < %s | FileCheck %s3; ARM64 AES intrinsic variants4 5define <16 x i8> @combineXorAeseZeroARM64(<16 x i8> %data, <16 x i8> %key) {6; CHECK-LABEL: define <16 x i8> @combineXorAeseZeroARM64(7; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {8; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> [[DATA]], <16 x i8> [[KEY]])9; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]10;11 %data.xor = xor <16 x i8> %data, %key12 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> %data.xor, <16 x i8> zeroinitializer)13 ret <16 x i8> %data.aes14}15 16define <16 x i8> @combineXorAeseZeroLhsARM64(<16 x i8> %data, <16 x i8> %key) {17; CHECK-LABEL: define <16 x i8> @combineXorAeseZeroLhsARM64(18; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {19; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> [[DATA]], <16 x i8> [[KEY]])20; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]21;22 %data.xor = xor <16 x i8> %data, %key23 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> zeroinitializer, <16 x i8> %data.xor)24 ret <16 x i8> %data.aes25}26 27define <16 x i8> @combineXorAeseNonZeroARM64(<16 x i8> %data, <16 x i8> %key) {28; CHECK-LABEL: define <16 x i8> @combineXorAeseNonZeroARM64(29; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {30; CHECK-NEXT: [[DATA_XOR:%.*]] = xor <16 x i8> [[DATA]], [[KEY]]31; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> [[DATA_XOR]], <16 x i8> splat (i8 -1))32; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]33;34 %data.xor = xor <16 x i8> %data, %key35 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8> %data.xor, <16 x i8> <i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1>)36 ret <16 x i8> %data.aes37}38 39define <16 x i8> @combineXorAesdZeroARM64(<16 x i8> %data, <16 x i8> %key) {40; CHECK-LABEL: define <16 x i8> @combineXorAesdZeroARM64(41; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {42; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> [[DATA]], <16 x i8> [[KEY]])43; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]44;45 %data.xor = xor <16 x i8> %data, %key46 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> %data.xor, <16 x i8> zeroinitializer)47 ret <16 x i8> %data.aes48}49 50define <16 x i8> @combineXorAesdZeroLhsARM64(<16 x i8> %data, <16 x i8> %key) {51; CHECK-LABEL: define <16 x i8> @combineXorAesdZeroLhsARM64(52; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {53; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> [[DATA]], <16 x i8> [[KEY]])54; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]55;56 %data.xor = xor <16 x i8> %data, %key57 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> zeroinitializer, <16 x i8> %data.xor)58 ret <16 x i8> %data.aes59}60 61define <16 x i8> @combineXorAesdNonZeroARM64(<16 x i8> %data, <16 x i8> %key) {62; CHECK-LABEL: define <16 x i8> @combineXorAesdNonZeroARM64(63; CHECK-SAME: <16 x i8> [[DATA:%.*]], <16 x i8> [[KEY:%.*]]) {64; CHECK-NEXT: [[DATA_XOR:%.*]] = xor <16 x i8> [[DATA]], [[KEY]]65; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> [[DATA_XOR]], <16 x i8> splat (i8 -1))66; CHECK-NEXT: ret <16 x i8> [[DATA_AES]]67;68 %data.xor = xor <16 x i8> %data, %key69 %data.aes = tail call <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8> %data.xor, <16 x i8> <i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1, i8 -1>)70 ret <16 x i8> %data.aes71}72 73declare <16 x i8> @llvm.aarch64.crypto.aese(<16 x i8>, <16 x i8>) #074declare <16 x i8> @llvm.aarch64.crypto.aesd(<16 x i8>, <16 x i8>) #075 76; SVE77 78define <vscale x 16 x i8> @combineXorAeseZeroLhsSVE(<vscale x 16 x i8> %data, <vscale x 16 x i8> %key) {79; CHECK-LABEL: define <vscale x 16 x i8> @combineXorAeseZeroLhsSVE(80; CHECK-SAME: <vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[KEY:%.*]]) {81; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aese(<vscale x 16 x i8> [[DATA]], <vscale x 16 x i8> [[KEY]])82; CHECK-NEXT: ret <vscale x 16 x i8> [[DATA_AES]]83;84 %data.xor = xor <vscale x 16 x i8> %data, %key85 %data.aes = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aese(<vscale x 16 x i8> zeroinitializer, <vscale x 16 x i8> %data.xor)86 ret <vscale x 16 x i8> %data.aes87}88 89define <vscale x 16 x i8> @combineXorAeseZeroRhsSVE(<vscale x 16 x i8> %data, <vscale x 16 x i8> %key) {90; CHECK-LABEL: define <vscale x 16 x i8> @combineXorAeseZeroRhsSVE(91; CHECK-SAME: <vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[KEY:%.*]]) {92; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aese(<vscale x 16 x i8> [[DATA]], <vscale x 16 x i8> [[KEY]])93; CHECK-NEXT: ret <vscale x 16 x i8> [[DATA_AES]]94;95 %data.xor = xor <vscale x 16 x i8> %data, %key96 %data.aes = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aese(<vscale x 16 x i8> %data.xor, <vscale x 16 x i8> zeroinitializer)97 ret <vscale x 16 x i8> %data.aes98}99 100define <vscale x 16 x i8> @combineXorAesdZeroLhsSVE(<vscale x 16 x i8> %data, <vscale x 16 x i8> %key) {101; CHECK-LABEL: define <vscale x 16 x i8> @combineXorAesdZeroLhsSVE(102; CHECK-SAME: <vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[KEY:%.*]]) {103; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aesd(<vscale x 16 x i8> [[DATA]], <vscale x 16 x i8> [[KEY]])104; CHECK-NEXT: ret <vscale x 16 x i8> [[DATA_AES]]105;106 %data.xor = xor <vscale x 16 x i8> %data, %key107 %data.aes = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aesd(<vscale x 16 x i8> zeroinitializer, <vscale x 16 x i8> %data.xor)108 ret <vscale x 16 x i8> %data.aes109}110 111define <vscale x 16 x i8> @combineXorAesdZeroRhsSVE(<vscale x 16 x i8> %data, <vscale x 16 x i8> %key) {112; CHECK-LABEL: define <vscale x 16 x i8> @combineXorAesdZeroRhsSVE(113; CHECK-SAME: <vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[KEY:%.*]]) {114; CHECK-NEXT: [[DATA_AES:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aesd(<vscale x 16 x i8> [[DATA]], <vscale x 16 x i8> [[KEY]])115; CHECK-NEXT: ret <vscale x 16 x i8> [[DATA_AES]]116;117 %data.xor = xor <vscale x 16 x i8> %data, %key118 %data.aes = tail call <vscale x 16 x i8> @llvm.aarch64.sve.aesd(<vscale x 16 x i8> %data.xor, <vscale x 16 x i8> zeroinitializer)119 ret <vscale x 16 x i8> %data.aes120}121 122declare <vscale x 16 x i8> @llvm.aarch64.sve.aese(<vscale x 16 x i8>, <vscale x 16 x i8>) #0123declare <vscale x 16 x i8> @llvm.aarch64.sve.aesd(<vscale x 16 x i8>, <vscale x 16 x i8>) #0124