Skip to content

Commit

Permalink
#0: REVERT-ME: disable falcon40b demo tests to reduce ci load
Browse files Browse the repository at this point in the history
Signed-off-by: Salar Hosseini <[email protected]>
  • Loading branch information
skhorasganiTT committed May 16, 2024
1 parent e491085 commit bb585d0
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/scripts/multi_chip/run_end_to_end_demos.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,11 @@ fi
cd $TT_METAL_HOME
export PYTHONPATH=$TT_METAL_HOME

# Falcon40B prefill 60 layer end to end with 10 loops; we need 8x8 grid size
WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/t3000/falcon40b/tests/ci/test_falcon_end_to_end_60_layer_t3000_prefill_10_loops.py
# # Falcon40B prefill 60 layer end to end with 10 loops; we need 8x8 grid size
# WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/t3000/falcon40b/tests/ci/test_falcon_end_to_end_60_layer_t3000_prefill_10_loops.py

# Falcon40B end to end demo (prefill + decode)
WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/t3000/falcon40b/tests/ci/test_falcon_end_to_end_t3000_demo_5_loops.py
# # Falcon40B end to end demo (prefill + decode)
# WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/t3000/falcon40b/tests/ci/test_falcon_end_to_end_t3000_demo_5_loops.py

# Falcon7B demo (perf verification and output verification)
pytest --disable-warnings -q -s --input-method=json --input-path='models/demos/t3000/falcon7b/input_data_t3000.json' models/demos/t3000/falcon7b/demo_t3000.py::test_demo_multichip[user_input0-8-True-perf_mode_stochastic_verify]
Expand Down

0 comments on commit bb585d0

Please sign in to comment.