diff --git a/KVM/qemu/tests/x86_cpuid.py b/KVM/qemu/tests/x86_cpuid.py new file mode 100644 index 0000000..0162f35 --- /dev/null +++ b/KVM/qemu/tests/x86_cpuid.py @@ -0,0 +1,143 @@ +#!/usr/bin/python3 + +# SPDX-License-Identifier: GPL-2.0-only +# Copyright (c) 2024 Intel Corporation + +# Author: Xudong Hao +# +# History: Aug. 2024 - Xudong Hao - creation + +import os +import sys +from avocado.utils import process +from avocado.core import exceptions +from virttest import error_context, env_process +from virttest import data_dir, asset +from virttest import utils_package + +curr_dir = os.path.dirname(os.path.abspath(__file__)) +check_dir = "%s/../../../BM/instruction-check" % curr_dir +sys.path.append(check_dir) +from feature_list import cpuid_info + + +def get_baremetal_dir(params): + """ + Get the test provider's BM absolute path. + :param params: Dictionary with the test parameters + """ + provider = params["provider"] + provider_info = asset.get_test_provider_info(provider) + if provider_info["uri"].startswith("file://"): + provider_dir = provider_info["uri"][7:] + else: + provider_dir = data_dir.get_test_provider_dir(provider) + baremetal_dir = os.path.join(provider_dir, "BM") + + return baremetal_dir + + +def prepare_cpuid(test, params, src_dir, vm=None, session=None): + """ + Compile cpuid test tool in host or guest. + Return the execuable test tool with absolute path. + :param test: QEMU test object + :param params: Dictionary with the test parameters + :param src_dir: Test tool srouce code directory of absolute path + :param vm: The vm object + :param session: Guest session + """ + source_file = params["source_file"] + exec_file = params["exec_file"] + src_cpuid = os.path.join(src_dir, source_file) + if session: + test_dir = params["test_dir"] + vm.copy_files_to(src_cpuid, test_dir) + if session.cmd_status("rpm -q gcc"): + if session.cmd_status("yum -y install gcc"): + test.cancel("Fail to install gcc in vm, please retest " + "this case again.") + else: + test_dir = src_dir + s, o = process.getstatusoutput("rpm -q gcc", shell=True) + if s != 0: + install_status = utils_package.package_install("gcc") + if not install_status: + test.cancel("Failed to install gcc") + + compile_cmd = "cd %s && gcc %s -o %s" % (test_dir, source_file, exec_file) + if session: + status = session.cmd_status(compile_cmd) + else: + status = process.system(compile_cmd, shell=True) + if status: + raise exceptions.TestError("Test suite compile failed.") + + return os.path.join(test_dir, exec_file) + + +def check_cpuid(params, feature, exec_bin, session=None): + """ + Run cpuid test in host or guest + :param params: Dictionary with the test parameters + :param feature: The feature to be checked + :param exec_bin: The execuable bianry tool with absolute path + :param session: Guest session + """ + args = cpuid_info[feature] + args_str = ' '.join(args) + + check_cmd = '%s %s' % (exec_bin, args_str) + func = process.getstatusoutput + if session: + func = session.cmd_status_output + s, o = func(check_cmd) + + return s + + +@error_context.context_aware +def run(test, params, env): + """ + Check cpuid inside guest. + 1) Check if current cpuid are support in host, if no, cancel test + 2) Boot up guest + 3) Check cpuid/s in guest + 4) Shutdown guest + :param test: QEMU test object + :param params: Dictionary with the test parameters + :param env: Dictionary with test environment. + """ + test_dir = params["test_dir"] + bm_dir = get_baremetal_dir(params) + src_dir = "%s/tools/cpuid_check" % bm_dir + + check_host_cpuid = params.get_boolean("check_host_cpuid") + if check_host_cpuid: + host_exec_bin = prepare_cpuid(test, params, src_dir) + + feature_names = params["features"] + for feature_name in feature_names.split(): + host_status = check_cpuid(params, feature_name, host_exec_bin) + if host_status: + process.system("rm %s -f" % host_exec_bin, shell=True, ignore_status=True) + test.cancel("Platform doesn't support %s" % feature_name) + test.log.info("Host cpuid check %s pass.", feature_names) + + try: + params["start_vm"] = "yes" + vm_name = params['main_vm'] + env_process.preprocess_vm(test, params, env, vm_name) + vm = env.get_vm(vm_name) + session = vm.wait_for_login() + + vm_exec_bin = prepare_cpuid(test, params, src_dir, vm, session) + for feature_name in feature_names.split(): + guest_status = check_cpuid(params, feature_name, vm_exec_bin, session) + if guest_status: + test.fail("%s cpuid check fail in guest." % feature_name) + test.log.info("Guest cpuid check %s pass.", feature_names) + finally: + process.system("rm %s -f" % host_exec_bin, shell=True, ignore_status=True) + session.cmd("rm %s/cpuid* -rf" % test_dir, ignore_all_errors=True) + session.close() diff --git a/KVM/qemu/x86_cpuid.cfg b/KVM/qemu/x86_cpuid.cfg new file mode 100644 index 0000000..0204d40 --- /dev/null +++ b/KVM/qemu/x86_cpuid.cfg @@ -0,0 +1,45 @@ +- x86_cpuid: + type = x86_cpuid + kill_vm_on_error = yes + start_vm = no + check_host_cpuid= yes + vga = std + test_dir = '/home' + source_file = 'cpuid_check.c' + exec_file = 'cpuid_check' + auto_cpu_model = "no" + cpu_model = host + variants: + - avx512_fp16: + # The name should be same as the KEY/s in cpuid_info dictionary + # "BM/instruction-check/feature_list.py" + features = "AVX512_FP16" + - serialize: + features = "SERIALIZE" + - tsxldtrk: + features = "TSXLDTRK" + - avx_vnni: + features = "AVX_VNNI" + - amx_bf16: + features = "AMX_BF16" + - amx_int8: + features = "AMX_INT8" + - amx_tile: + features = "AMX_TILE" + - amx_fp16: + features = "AMX_FP16" + - prefetchi: + features = "PREFETCHI" + - avx_ifma: + features = "AVX_IFMA" + - avx_vnni_int8: + features = "AVX_VNNI_INT8" + - avx_ne_convert: + features = "AVX_NE_CONVERT" + - cmpccxadd: + features = "CMPCCXADD" + variants: + - vm: + - tdvm: + machine_type_extra_params = "kernel-irqchip=split" + vm_secure_guest_type = tdx