[PATCH v6 23/26] selftest/x86/amx: Test cases for the AMX state management