import sys | |
# AMX tests must be enabled via build flag. | |
if not config.mlir_run_amx_tests: | |
config.unsupported = True | |
# No JIT on win32. | |
if sys.platform == 'win32': | |
config.unsupported = True | |
if config.intel_sde_executable: | |
# Run test in emulator (Intel SDE): AMX needs Sapphire Rapids CPU. | |
config.substitutions.append(('%lli', config.intel_sde_executable + ' -spr -- lli')) | |
else: | |
config.substitutions.append(('%lli', 'lli')) |