; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=riscv64 -mattr=+zknd -verify-machineinstrs < %s \
; RUN:   | FileCheck %s -check-prefix=RV64ZKND-ZKNE
; RUN: llc -mtriple=riscv64 -mattr=+zkne -verify-machineinstrs < %s \
; RUN:   | FileCheck %s -check-prefix=RV64ZKND-ZKNE

declare i64 @llvm.riscv.aes64ks2(i64, i64);

define i64 @aes64ks2(i64 %a, i64 %b) nounwind {
; RV64ZKND-ZKNE-LABEL: aes64ks2:
; RV64ZKND-ZKNE:       # %bb.0:
; RV64ZKND-ZKNE-NEXT:    aes64ks2 a0, a0, a1
; RV64ZKND-ZKNE-NEXT:    ret
    %val = call i64 @llvm.riscv.aes64ks2(i64 %a, i64 %b)
    ret i64 %val
}

declare i64 @llvm.riscv.aes64ks1i(i64, i32);

define i64 @aes64ks1i(i64 %a) nounwind {
; RV64ZKND-ZKNE-LABEL: aes64ks1i:
; RV64ZKND-ZKNE:       # %bb.0:
; RV64ZKND-ZKNE-NEXT:    aes64ks1i a0, a0, 10
; RV64ZKND-ZKNE-NEXT:    ret
    %val = call i64 @llvm.riscv.aes64ks1i(i64 %a, i32 10)
    ret i64 %val
}