configs: Add option to disable AVX in GPUFS

GPUFS+KVM simulations automatically enable AVX. This commit adds a
command line option to disable AVX if its not needed for a GPUFS
simulation.

Change-Id: Ic22592767dbdca86f3718eca9c837a8e29b6b781
This commit is contained in:
Vishnu Ramadas
2023-10-03 12:10:42 -05:00
parent f69191a31d
commit d3637a489d
2 changed files with 11 additions and 1 deletions

View File

@@ -234,7 +234,7 @@ def makeGpuFSSystem(args):
# If we are using KVM cpu, enable AVX. AVX is used in some ROCm libraries
# such as rocBLAS which is used in higher level libraries like PyTorch.
use_avx = False
if ObjectList.is_kvm_cpu(TestCPUClass):
if ObjectList.is_kvm_cpu(TestCPUClass) and not args.disable_avx:
# AVX also requires CR4.osxsave to be 1. These must be set together
# of KVM will error out.
system.workload.enable_osxsave = 1