diff --git a/.github/workflows/build_gpu-fs-sim.yml b/.github/workflows/build_gpu-fs-sim.yml index eb69627ad..6c8e65448 100644 --- a/.github/workflows/build_gpu-fs-sim.yml +++ b/.github/workflows/build_gpu-fs-sim.yml @@ -61,7 +61,7 @@ jobs: source /opt/intel/oneapi/setvars.sh set -x cd $HOME_DIR/frameworks.ai.mlir.mlir-extensions - cmake -S $BUILD_ROOT/llvm-project/llvm -B build -GNinja -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_PROJECTS=mlir -DLLVM_ENABLE_ASSERTIONS=ON -DLLVM_USE_LINKER=gold -DLLVM_ENABLE_ZSTD=OFF -DLLVM_EXTERNAL_PROJECTS="Imex" -DLLVM_EXTERNAL_IMEX_SOURCE_DIR=. -DIMEX_ENABLE_SYCL_RUNTIME=1 -DIMEX_ENABLE_L0_RUNTIME=1 -DLLVM_LIT_ARGS="-a -j 1 --debug --timeout=600 --filter=gemm_with_transposed_B_1kx1kx1k_f16_f16_f32.mlir" -DIMEX_ENABLE_FS_SIMULATOR=ON || exit + cmake -S $BUILD_ROOT/llvm-project/llvm -B build -GNinja -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_PROJECTS=mlir -DLLVM_ENABLE_ASSERTIONS=ON -DLLVM_USE_LINKER=gold -DLLVM_ENABLE_ZSTD=OFF -DLLVM_EXTERNAL_PROJECTS="Imex" -DLLVM_EXTERNAL_IMEX_SOURCE_DIR=. -DIMEX_ENABLE_SYCL_RUNTIME=1 -DIMEX_ENABLE_L0_RUNTIME=1 -DLLVM_LIT_ARGS="-a -j 1 --debug --timeout=1200 --filter=Integration/Dialect/XeGPU/*" -DIMEX_ENABLE_FS_SIMULATOR=ON || exit sed -e 's,-printAllFeatures,,g' -i ${WORK_ROOT}/fs/scripts/run_in_a_loop_rasty.sh export NEO_BINARIES_ROOT="$HOME"/neo (cd ${WORK_ROOT}/fs/scripts && source setup.sh && umd_driver_env_variables_export && ./run_in_a_loop_rasty.sh) &> fs-sim.log & diff --git a/test/Integration/Dialect/XeGPU/lit.local.cfg b/test/Integration/Dialect/XeGPU/lit.local.cfg index 28babf423..83ade2073 100644 --- a/test/Integration/Dialect/XeGPU/lit.local.cfg +++ b/test/Integration/Dialect/XeGPU/lit.local.cfg @@ -23,7 +23,9 @@ if(not config.imex_enable_pvc_target and not config.imex_enable_fs_simulator): slow_simulator_tests = [ 'gemm_4kx4kx4k_f16_f16_f16_w_8x32xf16_stores.mlir', - 'large_stores_8x32xf16_w_1d_vector_shuffle.vc.mlir', + 'gemm_4kx4kx4k_f16_f16_f16.mlir', + 'gemm_4kx4kx4k_f16_f16_f16_w_simple_B_prefetch.mlir', + 'flash_attention_fwd.mlir', ] if(config.imex_enable_fs_simulator):